diff --git a/Dockerfile b/Dockerfile
index 488800fc7239c0063145c269e0af052021fa146d..317f50d55c86c5283b2cfca1fd68f3d9672af978 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu22.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
@@ -52,7 +52,8 @@ 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 torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia
+    conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -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