Skip to content

Commit

Permalink
pip install coremltools onnx (ultralytics#2690)
Browse files Browse the repository at this point in the history
Requested in ultralytics#2686

(cherry picked from commit 9ccfa85)
  • Loading branch information
glenn-jocher authored and Lechtr committed Apr 12, 2021
1 parent 2c62c20 commit 066e000
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 066e000

Please sign in to comment.