tensorflow-haskell/docker/gpu/Dockerfile

# Prepare the image with:
#   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>"

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.3.0.tar.gz && \
    tar zxf libtensorflow-gpu-linux-x86_64-2.3.0.tar.gz -C /usr/local && \
    ldconfig

ENV LANG en_US.UTF-8