Chaging the pytorch version instead

Dockerfile

@@ -56,7 +56,7 @@ ENV PATH="$WRKSPCE/miniconda3/bin:${PATH}"
 RUN conda update -c defaults conda
 
 # Create the environment and install all the requirements
-RUN conda create -n segment_anything_env python=3.9 -y && \
+RUN conda create -n segment_anything_env python=3.8 -y && \
     conda init bash && \
     echo "conda activate segment_anything_env" >> ~/.bashrc
 
@@ -67,8 +67,9 @@ SHELL ["conda", "run", "-n", "segment_anything_env", "/bin/bash", "-c"]
 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==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch    
+    pip install opencv-python pycocotools matplotlib onnxruntime onnx 
+    #&& \
+RUN 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