From 533cf87dee4d13d6a67cffd1e19b4f8ad7313491 Mon Sep 17 00:00:00 2001 From: MStolen Date: Wed, 23 Oct 2024 13:56:14 -0500 Subject: [PATCH] Fix Tensorflow BURT Build --- TensorFlow/LanguageModeling/BERT/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorFlow/LanguageModeling/BERT/Dockerfile b/TensorFlow/LanguageModeling/BERT/Dockerfile index 7ad4404ca..32b57cd1e 100644 --- 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}