From 38e06e463574ba1102ebd9ccbc117312767500b5 Mon Sep 17 00:00:00 2001
From: Josh Meyer <joshua.richard.meyer@gmail.com>
Date: Thu, 10 Jun 2021 09:47:15 -0400
Subject: [PATCH] Use NVIDIA image in Dockerfile

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

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