grosse
/
jupyterlab-datascience-gpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

simple tf
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
maltegrosse 2023-01-30 15:56:38 +01:00
parent 637c0fb0f7
commit 361051ad4d
1 changed files with 1 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Dockerfile
View File

@ -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
RUN conda install -c conda-forge tensorflow-gpu=2.10 -y