mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update Dockerfile-arm64 (#10918)

Browse Source 
Docker fixes
This commit is contained in:
Glenn Jocher 2023-02-07 02:15:54 +04:00 committed by GitHub
parent 9ba18266b2
commit c3c8e71d7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/docker/Dockerfile-arm64
View File

@ -18,8 +18,8 @@ RUN apt install --no-install-recommends -y python3-pip git zip curl htop gcc lib
# Install pip packages
COPY requirements.txt .
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache -r requirements.txt albumentations comet gsutil notebook \
coremltools onnx onnxruntime 'openvino-dev>=2022.3'
RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \
coremltools onnx onnxruntime
# tensorflow-aarch64 tensorflowjs \
# Create working directory

2
utils/docker/Dockerfile-cpu
View File

@ -18,7 +18,7 @@ RUN apt install --no-install-recommends -y python3-pip git zip curl htop libgl1-
# Install pip packages
COPY requirements.txt .
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 gsutil notebook \
coremltools onnx onnx-simplifier onnxruntime 'openvino-dev>=2022.3' \
# tensorflow tensorflowjs \
--extra-index-url https://download.pytorch.org/whl/cpu