diff --git a/Dockerfile.train b/Dockerfile.train
index a31f17b3..0e86fa2a 100644
--- a/Dockerfile.train
+++ b/Dockerfile.train
@@ -44,7 +44,7 @@ RUN wget --no-check-certificate https://github.com/coqui-ai/STT/releases/downloa
     rm temp.tar.xz
 
 
-FROM nvcr.io/nvidia/tensorflow:21.05-tf1-py3
+FROM nvcr.io/nvidia/tensorflow:20.06-tf1-py3
 ENV DEBIAN_FRONTEND=noninteractive
 
 # We need to purge python3-xdg because