Update dockerfiles

pillargg · Jul 15, 2021 · 1317e85 · 1317e85
1 parent f323191
commit 1317e85
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/Dockerfile.arm64 b/Dockerfile.arm64
@@ -1,10 +1,9 @@
-FROM chand1012/tensorflow:2.5.0-py3
+FROM chand1012/tensorflow:2.5.0
 
 RUN mkdir /gpt-2
 WORKDIR /gpt-2
 ADD . /gpt-2
 RUN pip3 install -r requirements.txt
 RUN python3 download_model.py 124M
 RUN python3 download_model.py 355M
-RUN python3 download_model.py 774M
-RUN python3 download_model.py 1558M
+# additional models would not build on the Pi
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.5.0-py3
+FROM tensorflow/tensorflow:2.5.0
 
 ENV LANG=C.UTF-8
 RUN mkdir /gpt-2

diff --git a/Dockerfile.gpu b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.5.0-gpu-py3
+FROM tensorflow/tensorflow:2.5.0-gpu
 
 # nvidia-docker 1.0
 LABEL com.nvidia.volumes.needed="nvidia_driver"