From 066e00087c2ec3995c3215c97ce3d862a8f5b162 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 2 Apr 2021 13:00:46 +0200 Subject: [PATCH] pip install coremltools onnx (#2690) Requested in https://github.com/ultralytics/yolov5/issues/2686 (cherry picked from commit 9ccfa85249a2409d311bdf2e817f99377e135091) --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index c0484e5b9c1c..b47e5bbff194 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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