Specify TensorFlow version (2.3.0) in Dockerfile

This commit is contained in:
Bart Schuurmans 2023-01-17 12:23:02 +01:00 committed by fkm3
parent 86a4a12aa8
commit b5e3a98870
3 changed files with 3 additions and 3 deletions

View File

@ -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

View File

@ -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 <tensorflow-haskell@googlegroups.com>"

View File

@ -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 <tensorflow-haskell@googlegroups.com>"