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"