FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d # debian:bullseye-20230109-slim ARG NB_USER="jovyan" ARG NB_UID="1000" ARG NB_GID="100" ENV CONDA_CUDA_OVERRIDE="12.1" ENV NUM_GPUS=1 USER root RUN apt-get update --yes && \ apt-get install --yes --no-install-recommends \ ffmpeg \ libportaudio2 && \ apt-get clean && rm -rf /var/lib/apt/lists/* USER $NB_UID RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia