pip install coremltools onnx (#2690)

Requested in https://github.com/ultralytics/yolov5/issues/2686
This commit is contained in:
Glenn Jocher 2021-04-02 13:00:46 +02:00 committed by GitHub
parent 17300a4c7b
commit 9ccfa85249
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -8,7 +8,7 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
COPY requirements.txt . COPY requirements.txt .
RUN python -m pip install --upgrade pip RUN python -m pip install --upgrade pip
RUN pip uninstall -y nvidia-tensorboard nvidia-tensorboard-plugin-dlprof 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 # Create working directory
RUN mkdir -p /usr/src/app RUN mkdir -p /usr/src/app