Pin Torch (was auto bumped to 2.2)

ucsd-ets · Jan 31, 2024 · 8481d13 · 8481d13
1 parent 6f11986
commit 8481d13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -80,7 +80,7 @@ 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 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
+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 && \