cuda 12
ci/woodpecker/push/woodpecker Pipeline was successful Details

This commit is contained in:
Malte Grosse 2024-03-18 13:56:40 +09:00
parent 036a23cecb
commit 68e7fe63cb
1 changed files with 2 additions and 2 deletions

View File

@ -3,6 +3,6 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d
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 CONDA_CUDA_OVERRIDE="12.0"
ENV NUM_GPUS=1 ENV NUM_GPUS=1
RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge RUN CONDA_OVERRIDE_CUDA="12.0" conda install tensorflow-gpu==2.15.0 -c conda-forge