1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-23 03:19:44 +01:00

Bump TF to 2.17

Signed-off-by: Mihai Maruseac <mihaimaruseac@google.com>
This commit is contained in:
Mihai Maruseac 2024-08-15 08:09:06 -07:00
parent 925c2e9515
commit 26f6439b73
No known key found for this signature in database
5 changed files with 16 additions and 16 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 git clone --recursive https://github.com/tensorflow/haskell.git tensorflow-haskell
cd tensorflow-haskell cd tensorflow-haskell
docker build -t tensorflow/haskell:2.12.0 docker docker build -t tensorflow/haskell:2.17.0 docker
# TODO: move the setup step to the docker script. # TODO: move the setup step to the docker script.
stack --docker setup stack --docker setup
stack --docker test stack --docker test
@ -90,7 +90,7 @@ stack --docker build --exec Main
If you want to use GPU you can do: If you want to use GPU you can do:
``` ```
IMAGE_NAME=tensorflow/haskell:2.12.0-gpu IMAGE_NAME=tensorflow/haskell:2.17.0-gpu
docker build -t $IMAGE_NAME docker/gpu docker build -t $IMAGE_NAME docker/gpu
# TODO: move the setup step to the docker script. # TODO: move the setup step to the docker script.
stack --docker --docker-image=$IMAGE_NAME setup stack --docker --docker-image=$IMAGE_NAME setup

View file

@ -3,7 +3,7 @@
# stack to be installed on the host. This comes at the expense of # stack to be installed on the host. This comes at the expense of
# flexibility. # flexibility.
FROM tensorflow/tensorflow:2.12.0 FROM tensorflow/tensorflow:2.17.0
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>" LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
# The build context directory is the top of the tensorflow-haskell # The build context directory is the top of the tensorflow-haskell
@ -36,8 +36,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \ curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \ unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \ chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz && \ curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.17.0.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \ tar zxf libtensorflow-cpu-linux-x86_64-2.17.0.tar.gz -C /usr/local && \
ldconfig && \ ldconfig && \
stack setup && \ stack setup && \
stack test --only-dependencies stack test --only-dependencies

View file

@ -1,6 +1,6 @@
# Prepare the image with: # Prepare the image with:
# docker build -t tensorflow/haskell:2.12.0 docker # docker build -t tensorflow/haskell:2.17.0 docker
FROM tensorflow/tensorflow:2.12.0 FROM tensorflow/tensorflow:2.17.0
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>" LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
RUN apt-get update RUN apt-get update
@ -31,8 +31,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \ curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \ unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \ chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz && \ curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.17.0.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \ tar zxf libtensorflow-cpu-linux-x86_64-2.17.0.tar.gz -C /usr/local && \
ldconfig ldconfig
ENV LANG en_US.UTF-8 ENV LANG en_US.UTF-8

View file

@ -1,6 +1,6 @@
# Prepare the image with: # Prepare the image with:
# docker build -t tensorflow/haskell:2.12.0-gpu docker/gpu # docker build -t tensorflow/haskell:2.17.0-gpu docker/gpu
FROM tensorflow/tensorflow:2.12.0-gpu FROM tensorflow/tensorflow:2.17.0-gpu
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>" LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
RUN apt-get update RUN apt-get update
@ -31,8 +31,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \ curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \ unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \ chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz && \ curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.17.0.tar.gz && \
tar zxf libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \ tar zxf libtensorflow-gpu-linux-x86_64-2.17.0.tar.gz -C /usr/local && \
ldconfig ldconfig
ENV LANG en_US.UTF-8 ENV LANG en_US.UTF-8

View file

@ -29,10 +29,10 @@ extra-include-dirs:
docker: docker:
enable: false enable: false
image: tensorflow/haskell:2.12.0 image: tensorflow/haskell:2.17.0
nix: nix:
enable: false enable: false
# nixos-22.11 with libtensorflow 2.10.1 # nixos-24.05 with libtensorflow 2.13.0
path: ["nixpkgs=https://github.com/NixOS/nixpkgs/archive/2dea8991d89b9f1e78d874945f78ca15f6954289.tar.gz"] path: ["nixpkgs=https://github.com/NixOS/nixpkgs/archive/63dacb46bf939521bdc93981b4cbb7ecb58427a0.tar.gz"]
packages: [snappy, zlib, protobuf, libtensorflow] packages: [snappy, zlib, protobuf, libtensorflow]