Trying older pytorch instead

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"]