From d2d6071a6f7addefd324e5ed2546b9da72c1b387 Mon Sep 17 00:00:00 2001
From: ct00659um <ct00659@surrey.ac.uk>
Date: Fri, 10 Nov 2023 15:42:29 +0000
Subject: [PATCH] Trying older pytorch instead
---
Dockerfile | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/Dockerfile b/Dockerfile
index 1c59f20..cc4f92f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -68,7 +68,9 @@ RUN pip install setuptools==60.9.3 && \
git clone https://github.com/facebookresearch/segment-anything.git && \
cd segment-anything && pip install -e . && \
pip install opencv-python pycocotools matplotlib onnxruntime onnx && \
- conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
+ conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
+#conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
+ #conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
# Set the default command to bash to activate conda environment on run
CMD ["/bin/bash"]
--
GitLab