From 027dd257bb6571eef86d8d60cac037cdcce1b2c1 Mon Sep 17 00:00:00 2001
From: 85zhanghao <94844844+85zhanghao@users.noreply.github.com>
Date: Thu, 23 Jun 2022 11:21:23 +0800
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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