diff --git a/Dockerfile b/Dockerfile
index 5957661..e7cd850 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM  pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
+FROM  pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel
 RUN   pip  install addict  future   lmdb   matplotlib   numpy
 RUN   pip  install  Pillow  pyyaml  requests  scikit-image
 RUN   pip  install  scikit-image  scipy  tb-nightly   tqdm   yapf