diff --git a/Dockerfile b/Dockerfile index d833780..03e5651 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,25 +1,5 @@ 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 \ No newline at end of file +RUN conda install -c conda-forge tensorflow-gpu=2.10 -y \ No newline at end of file