FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:latest

USER root
RUN apt-get update && apt upgrade -y && apt-get install -y --no-install-recommends \
    build-essential gnupg2 curl ca-certificates && \
    curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb && \
    dpkg -i cuda-keyring_1.0-1_all.deb && \
    apt-get purge --autoremove -y curl \
    && rm -rf /var/lib/apt/lists/*

RUN apt-get update && apt-get install -y --no-install-recommends \ 
    cuda-11-8 cuda-cudart-11-8 nvidia-gds-11-8 cuda-toolkit-11-8 cuda-libraries-11-8

#RUN wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/cudnn-local-repo-ubuntu2204-8.7.0.84_1.0-1_amd64.deb
#RUN dpkg -i cudnn-local-repo-ubuntu2204-8.7.0.84_1.0-1_amd64.deb && rm -f cudnn-local-repo-ubuntu2204-8.7.0.84_1.0-1_amd64.deb
RUN apt update && apt install -y libcudnn8

#RUN pip install tensorflow
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
RUN pip install --upgrade tensorrt
USER jovyan

# https://anaconda.org/pytorch-test/pytorch-cuda
#https://github.com/NVIDIA/nvidia-docker/issues/871#issuecomment-966668913