From d87ab88286b7d524a07e22dfc44705b3f453c4bb Mon Sep 17 00:00:00 2001 From: maltegrosse Date: Wed, 1 Feb 2023 16:34:23 +0100 Subject: [PATCH] env --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index 968dd6f..2532d17 100644 --- a/Dockerfile +++ b/Dockerfile @@ -100,6 +100,7 @@ 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 export CONDA_CUDA_OVERRIDE="11.2" #RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge #==2.11.0=cuda* RUN mamba install -y -c conda-forge tensorflow-gpu