Specify TensorFlow version (2.3.0) in Dockerfile

LumiGuide · Jan 20, 2023 · b5e3a98 · b5e3a98
1 parent 86a4a12
commit b5e3a98
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -72,7 +72,7 @@ working, the following commands will compile and run the tests.
 ```
 git clone --recursive https://github.com/tensorflow/haskell.git tensorflow-haskell
 cd tensorflow-haskell
-IMAGE_NAME=tensorflow/haskell:v0
+IMAGE_NAME=tensorflow/haskell:2.3.0
 docker build -t $IMAGE_NAME docker
 # TODO: move the setup step to the docker script.
 stack --docker --docker-image=$IMAGE_NAME setup

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,5 +1,5 @@
 # Prepare the image with:
-#   docker build -t tensorflow/haskell:v0 docker
+#   docker build -t tensorflow/haskell:2.3.0 docker
 FROM tensorflow/tensorflow:2.3.0
 LABEL maintainer="TensorFlow authors <[email protected]>"
 

diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile
@@ -1,5 +1,5 @@
 # Prepare the image with:
-#   docker build -t tensorflow/haskell:1.14.0-gpu docker/gpu
+#   docker build -t tensorflow/haskell:2.3.0-gpu docker/gpu
 FROM tensorflow/tensorflow:2.3.0-gpu
 LABEL maintainer="TensorFlow authors <[email protected]>"