diff --git a/Dockerfile b/Dockerfile
index 1c59f209f0c19d697b2d91ca765f57ab54eaa86a..cc4f92f38f3f0164742bf9c73ef046f01633da16 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"]