Browse Source

pip install coremltools onnx (#2690)

Requested in https://github.com/ultralytics/yolov5/issues/2686
5.0
Glenn Jocher GitHub 3 years ago
parent
commit
9ccfa85249
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      Dockerfile

+ 1
- 1
Dockerfile View File

@@ -8,7 +8,7 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
COPY requirements.txt .
RUN python -m pip install --upgrade pip
RUN pip uninstall -y nvidia-tensorboard nvidia-tensorboard-plugin-dlprof
RUN pip install --no-cache -r requirements.txt gsutil notebook
RUN pip install --no-cache -r requirements.txt coremltools onnx gsutil notebook

# Create working directory
RUN mkdir -p /usr/src/app

Loading…
Cancel
Save