diff --git a/Dockerfile b/Dockerfile
index aa01174568c789ad307bf7f65e5757730bda9a8d..488800fc7239c0063145c269e0af052021fa146d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -51,7 +51,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
+    pip install opencv-python pycocotools matplotlib onnxruntime onnx && \
+    conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
 
 # Uncomment the following line if you want to install PyTorch and related packages within the Conda environment
 # RUN conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
@@ -59,6 +60,10 @@ RUN pip install setuptools==60.9.3 && \
 # Set the default command to bash to activate conda environment on run
 CMD ["/bin/bash"]
 
+# RUN conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
+
+# RUN conda activate segment_anything_env
+
 
 # RUN conda install pytorch>=1.7 torchvision>=0.8 torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
 #RUN pip3 install ffmpeg gnutls