1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-12-25 19:19:45 +01:00
tensorflow-haskell/docker/gpu/Dockerfile
Jeroen Bransen 925c2e9515
Update to TensorFlow 2.12 (#292)
* Fix compilation with tensorflow 2.12 and higher

* Update tensorflow submodule to 2.12.0

* Update Dockerfiles to 2.12.0
2023-08-14 20:36:35 -07:00

38 lines
1.3 KiB
Docker

# Prepare the image with:
# docker build -t tensorflow/haskell:2.12.0-gpu docker/gpu
FROM tensorflow/tensorflow:2.12.0-gpu
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
RUN apt-get update
RUN apt-get install -y \
# Required by snappy-frames dependency.
libsnappy-dev \
# Avoids /usr/bin/ld: cannot find -ltinfo
libncurses5-dev \
# Makes stack viable in the container
libgmp-dev \
zlib1g-dev \
# For fetching dependencies
curl \
unzip \
# Required for locales configuration.
locales \
# Required for tcp connections by stack (See: https://github.com/tensorflow/haskell/issues/182)
netbase
# Support output of Unicode characters.
RUN dpkg-reconfigure locales && \
locale-gen en_US.UTF-8 && \
update-locale LANG=en_US.UTF-8
# Installs protoc and the libraries.
RUN \
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 && \
chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz && \
tar zxf libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \
ldconfig
ENV LANG en_US.UTF-8