From f907fb65304f3a9962f3f2e8cedc4b5a021e34ce Mon Sep 17 00:00:00 2001 From: Chandler Lofland Date: Wed, 14 Jul 2021 10:12:30 -0400 Subject: [PATCH] Update tensorflow version --- Dockerfile.cpu | 2 +- Dockerfile.gpu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.cpu b/Dockerfile.cpu index b6e4f9496..bab941069 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:1.12.0-py3 +FROM tensorflow/tensorflow:1.13.1-py3 ENV LANG=C.UTF-8 RUN mkdir /gpt-2 diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 5ac049aff..57d8d516d 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:1.12.0-gpu-py3 +FROM tensorflow/tensorflow:1.13.1-gpu-py3 # nvidia-docker 1.0 LABEL com.nvidia.volumes.needed="nvidia_driver"