Update Dockerfile

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 && \