diff --git a/Dockerfile b/Dockerfile
index 91da5878..7aebe10a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:19.12-py3
+FROM nvcr.io/nvidia/pytorch:20.01-py3
 
 # Install dependencies (pip or conda)
 RUN pip install -U gsutil