8 lines
283 B
Docker
8 lines
283 B
Docker
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 |