diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile index 4ecc2d52..85a87c97 100644 --- a/images/scipy-ml-notebook/Dockerfile +++ b/images/scipy-ml-notebook/Dockerfile @@ -80,7 +80,10 @@ RUN pip install datascience \ # We already have the lib files imported into LD_LIBRARY_PATH by CUDDN and the cudatoolkit. let's remove these and save some image space. # Beware of potentially needing to update these if we update the drivers. -RUN pip install torch==2.2.0 torchvision torchaudio typing-extensions==4.5.0 --index-url https://download.pytorch.org/whl/cu118 && \ + +# torch 2.2.0 requires typing-extensions 4.8+, conflicts with tensorflow 2.13.*. won't be able to upgrade torch until we upgrade tf +# defer until we have debian nodes +RUN pip install torch==2.1.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \ pip cache purge && \ rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcudnn_cnn_infer.so.8 && \ rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcublasLt.so.11 && \