diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile index 1a042ba4..bde6804e 100644 --- a/images/scipy-ml-notebook/Dockerfile +++ b/images/scipy-ml-notebook/Dockerfile @@ -75,7 +75,7 @@ RUN pip install --no-cache-dir datascience \ ARG TORCH_VER="2.0.1+cu118" # torch must be installed separately since it requires a non-pypi repo. See stable version above -RUN pip install install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \ +RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \ pip cache purge ### UNCOMMENT WHEN WE HAVE NVIDIA 11.8 ON NODES