From f0dfb7b1d8d8fa70819a5a371e2b4a867c3bac5c Mon Sep 17 00:00:00 2001
From: chand1012 <chandler@chand1012.dev>
Date: Wed, 14 Jul 2021 15:57:39 -0400
Subject: [PATCH] Bump tensorflow version

---
 Dockerfile.cpu | 2 +-
 Dockerfile.gpu | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index bab941069..9bf0c6144 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.13.1-py3
+FROM tensorflow/tensorflow:1.15.5-py3
 
 ENV LANG=C.UTF-8
 RUN mkdir /gpt-2
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 57d8d516d..97c8c9d89 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.13.1-gpu-py3
+FROM tensorflow/tensorflow:1.15.5-gpu-py3
 
 # nvidia-docker 1.0
 LABEL com.nvidia.volumes.needed="nvidia_driver"