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="11.2" ENV NUM_GPUS=1 RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu