Update Dockerfile

85zhanghao · Jun 23, 2022 · 027dd25 · 027dd25
1 parent 8cd96c7
commit 027dd25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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