diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 5437f44..1036445 100755 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:latest-py3 +FROM tensorflow/tensorflow:1.12.0-py3 RUN apt-get update -qq -y \ && apt-get install -y libsm6 libxrender1 libxext-dev python3-tk\ diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 1692d1f..99041d0 100755 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM tensorflow/tensorflow:latest-gpu-py3 +FROM tensorflow/tensorflow:1.12.0-gpu-py3 RUN apt-get update -qq -y \ && apt-get install -y libsm6 libxrender1 libxext-dev python3-tk\