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

add torch
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Malte Grosse 2024-03-18 10:30:04 +09:00
parent 6cbee7a51e
commit 1c2ed44563
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -3,3 +3,6 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12c
ARG NB_USER="jovyan" ARG NB_USER="jovyan"
ARG NB_UID="1000" ARG NB_UID="1000"
ARG NB_GID="100" 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