Update Dockerfile (#10917)
* Update Dockerfile Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update Dockerfile-arm64 Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
90f23519c8
commit
9ba18266b2
2 files changed
+1
-9
No files matched your search
@@ -31,14 +31,6 @@ RUN pip install --no-cache -r requirements.txt albumentations comet gsutil noteb
|
|||||||
coremltools onnx onnx-simplifier onnxruntime 'openvino-dev>=2022.3'
|
coremltools onnx onnx-simplifier onnxruntime 'openvino-dev>=2022.3'
|
||||||
# tensorflow tensorflowjs \
|
# tensorflow tensorflowjs \
|
||||||
|
|
||||||
# Create working directory
|
|
||||||
RUN mkdir -p /usr/src/app
|
|
||||||
WORKDIR /usr/src/app
|
|
||||||
|
|
||||||
# Copy contents
|
|
||||||
# COPY . /usr/src/app (issues as not a .git directory)
|
|
||||||
RUN git clone https://github.com/ultralytics/yolov5 /usr/src/app
|
|
||||||
|
|
||||||
# Set environment variables
|
# Set environment variables
|
||||||
ENV OMP_NUM_THREADS=1
|
ENV OMP_NUM_THREADS=1
|
||||||
|
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ RUN apt install --no-install-recommends -y python3-pip git zip curl htop gcc lib
|
|||||||
COPY requirements.txt .
|
COPY requirements.txt .
|
||||||
RUN python3 -m pip install --upgrade pip wheel
|
RUN python3 -m pip install --upgrade pip wheel
|
||||||
RUN pip install --no-cache -r requirements.txt albumentations comet gsutil notebook \
|
RUN pip install --no-cache -r requirements.txt albumentations comet gsutil notebook \
|
||||||
coremltools onnx onnx-simplifier onnxruntime 'openvino-dev>=2022.3'
|
coremltools onnx onnxruntime 'openvino-dev>=2022.3'
|
||||||
# tensorflow-aarch64 tensorflowjs \
|
# tensorflow-aarch64 tensorflowjs \
|
||||||
|
|
||||||
# Create working directory
|
# Create working directory
|
||||||
|
|||||||
Reference in new issue
Block a user