Dockerfile for gt4py cuda

Dockerfile

FROM nvidia/cuda:10.2-devel-ubuntu18.04
ARG CUDAVER=10.2
ARG CUDAVER_ND=cuda102
ENV LANG C.UTF-8
ENV LC_ALL C.UTF-8

ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update -qq && apt-get install -qq -y --no-install-recommends \
        strace \
        build-essential \
        cuda-toolkit-$CUDAVER \
        tar \
        wget \
        curl \
        ca-certificates \
        zlib1g-dev \
        libssl-dev \
        libbz2-dev \
        libsqlite3-dev \
        llvm \
        libncurses5-dev \
        libncursesw5-dev \
        xz-utils \
        tk-dev \
        libffi-dev \
        liblzma-dev \
        python-openssl \
        python3 \
        python3-dev \
        python3-venv \
        python3-numpy \
        libreadline-dev \
        git \
        htop && \
    apt-get clean

ENV CUDA_HOME /usr/local/cuda

WORKDIR /root
RUN python3 -m venv venv && \
    . venv/bin/activate && \
    python -m pip install --upgrade wheel && \
    git clone -b cxx_cuda_backends https://github.com/eddie-c-davis/gt4py && \
    pip install -e ./gt4py[$CUDAVER_ND]
RUN echo '. /root/venv/bin/activate' >> /root/.bashrc

# https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker
# sudo docker build . -t gt4py
# sudo docker run --rm --gpus all -it --cap-add=SYS_ADMIN gt4py