diff --git a/Dockerfile b/Dockerfile
index 65cff00..61f259d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,8 @@
-FROM mingzailao/cuda:11.3.1-cudnn8-devel-ubuntu20.04
+FROM mingzailao/cuda:10.2-cudnn8-devel-ubuntu18.04
 
-RUN pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu113
+
+
+RUN pip3 install torch==1.9.1+cu102  --extra-index-url https://download.pytorch.org/whl/cu102