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

gpu
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
maltegrosse 2023-02-01 15:58:26 +01:00
parent d5b1e596d1
commit f02fc16014
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.woodpecker.yml
View File

@ -5,6 +5,7 @@ pipeline:
group: docker
environment:
- CONDA_CUDA_OVERRIDE=11.2
- NVIDIA_VISIBLE_DEVICES=all
settings:
registry: https://git.sandbox.iuk.hdm-stuttgart.de
repo: git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience-gpu

3
Dockerfile
View File

@ -100,7 +100,8 @@ RUN set -x && \
#RUN CONDA_CUDA_OVERRIDE="11.2" mamba install --yes -c conda-forge tensorflow==2.10.0=cuda112py310*
ENV CONDA_CUDA_OVERRIDE="11.2"
RUN mamba install -y tensorflow==2.11.0=cuda112py310* cudatoolkit>=11.2 -c conda-forge
RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
#==2.11.0=cuda*
#RUN mamba install -y -c conda-forge tensorflow-gpu
WORKDIR /tmp
RUN mamba install --quiet --yes \