From 68e7fe63cb49b4e833ccd795a2b596ee854a408d Mon Sep 17 00:00:00 2001 From: Malte Grosse Date: Mon, 18 Mar 2024 13:56:40 +0900 Subject: [PATCH] cuda 12 --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 124fa50..bb25878 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,6 +3,6 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d ARG NB_USER="jovyan" ARG NB_UID="1000" ARG NB_GID="100" -ENV CONDA_CUDA_OVERRIDE="11.2" +ENV CONDA_CUDA_OVERRIDE="12.0" ENV NUM_GPUS=1 -RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge \ No newline at end of file +RUN CONDA_OVERRIDE_CUDA="12.0" conda install tensorflow-gpu==2.15.0 -c conda-forge \ No newline at end of file