From 78339343719fc3d175b7a0a0c80361ab9fe32826 Mon Sep 17 00:00:00 2001
From: jenhaoyang <steven05yang@gmail.com>
Date: Fri, 12 Mar 2021 14:58:58 +0800
Subject: [PATCH] Fix python required version > 3.6 bug

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

diff --git a/Dockerfile b/Dockerfile
index 2b6626a2..8ffc2c15 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
+FROM pytorch/pytorch:1.8.0-cuda11.1-cudnn8-runtime
 
 # install torchelastic
 WORKDIR /opt/torchelastic