25 lines
1.2 KiB
Docker
25 lines
1.2 KiB
Docker
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
|
|
RUN pip install pytorch
|
|
USER jovyan
|
|
|
|
# https://anaconda.org/pytorch-test/pytorch-cuda
|
|
#https://github.com/NVIDIA/nvidia-docker/issues/871#issuecomment-966668913 |