tensorflow==1.15.2

Tensorflow 2 is incompatible with this script. 1.15.2 is the most recent release of Tensorflow 1.
Hugo0 · Feb 1, 2020 · c4745e6 · c4745e6
1 parent ef992f4
commit c4745e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -8,7 +8,7 @@ COPY checkpoint /checkpoint
 # Make changes to the requirements/app here.
 # This Dockerfile order allows Docker to cache the checkpoint layer
 # and improve build times if making changes.
-RUN pip3 --no-cache-dir install tensorflow gpt-2-simple starlette uvicorn ujson
+RUN pip3 --no-cache-dir install tensorflow==1.15.2 gpt-2-simple starlette uvicorn ujson
 COPY app.py /
 
 # Clean up APT when done.