Fix Tensorflow BURT Build

NVIDIA · Oct 23, 2024 · 533cf87 · 533cf87
1 parent 729963d
commit 533cf87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TensorFlow/LanguageModeling/BERT/Dockerfile b/TensorFlow/LanguageModeling/BERT/Dockerfile
@@ -1,4 +1,4 @@
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.06-tf1-py3
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:21.07-tf1-py3
 
 FROM ${FROM_IMAGE_NAME}