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

tf
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Malte Grosse 2024-03-18 12:46:25 +09:00
parent 82acfa910f
commit 036a23cecb
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.woodpecker.yml
View File

@ -7,7 +7,7 @@ steps:
settings:
registry: git.sandbox.iuk.hdm-stuttgart.de
repository: grosse/jupyterlab-datascience-gpu
tag: 4.0.x-gpu-torch
tag: 4.0.x-gpu-tf
architectures: amd64
context: Dockerfile
loglevel: debug

2
Dockerfile
View File

@ -5,4 +5,4 @@ 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
RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge