jupyterlab-datascience-gpu/Dockerfile

8 lines
283 B
Docker
Raw Normal View History

2024-03-18 01:33:31 +00:00
FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12b
2023-02-01 09:05:43 +00:00
# debian:bullseye-20230109-slim
ARG NB_USER="jovyan"
ARG NB_UID="1000"
2024-03-18 01:30:04 +00:00
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