From 307750e547e575b4bf6940147d574c6e40dc9af3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 26 Jun 2021 16:51:10 +0200 Subject: [PATCH] Update Dockerfile --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 4754801b5b56..a5b9da06a3d5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,7 +9,8 @@ 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 coremltools onnx gsutil notebook -RUN pip install --no-cache torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html +RUN pip install --no-cache -U torch torchvision numpy +# RUN pip install --no-cache torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html # Create working directory RUN mkdir -p /usr/src/app