diff --git a/Dockerfile b/Dockerfile
index c01b88328c361200140dd988db6600181440a021..68f16c158c159a312f0c4379963afb463b01871b 100644
--- a/Dockerfile
+++ b/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.8 -y && \
+RUN conda create -n segment_anything_env python=3.9 -y && \
     conda init bash && \
     echo "conda activate segment_anything_env" >> ~/.bashrc
 
@@ -67,8 +67,8 @@ 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    
+    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 
 #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