diff --git a/Dockerfile b/Dockerfile
index 12b3a7e18a7095d23487aef69c77aaa510b15289..eea875d1a9982137b540e27c97e13c79f4edab5c 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.9 -y && \
+RUN conda create -n segment_anything_env python=3.8 -y && \
     conda init bash && \
     echo "conda activate segment_anything_env" >> ~/.bashrc
 
@@ -64,11 +64,13 @@ RUN conda create -n segment_anything_env python=3.9 -y && \
 SHELL ["conda", "run", "-n", "segment_anything_env", "/bin/bash", "-c"]
 
 # Install the requirements
-RUN pip install setuptools==60.9.3 && \
+RUN conda activate segment_anything_env && \
+    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 
+
 #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