diff --git a/Dockerfile.train b/Dockerfile.train
index e9244dde..0d103559 100644
--- a/Dockerfile.train
+++ b/Dockerfile.train
@@ -1,6 +1,6 @@
 # Please refer to the TRAINING documentation, "Basic Dockerfile for training"
 
-FROM tensorflow/tensorflow:1.15.4-gpu-py3
+FROM nvcr.io/nvidia/tensorflow:21.05-tf1-py3
 ENV DEBIAN_FRONTEND=noninteractive
 
 # We need to purge python3-xdg because it's breaking STT install later with