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