diff --git a/cpu.Dockerfile b/cpu.Dockerfile
index 72a6726..1fa47bf 100644
--- a/cpu.Dockerfile
+++ b/cpu.Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.5.0
+FROM tensorflow/tensorflow:1.8.0-py3
 
 RUN mkdir /root/mimic2
 COPY . /root/mimic2
diff --git a/gpu.Dockerfile b/gpu.Dockerfile
index ff72a5b..96d0be0 100644
--- a/gpu.Dockerfile
+++ b/gpu.Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:1.5.0-gpu
+FROM tensorflow/tensorflow:1.8.0-gpu-py3
 
 RUN mkdir /root/mimic2
 COPY . /root/mimic2