From d5b1e596d184bdc4fbf4fadabe69f3792fbaaf08 Mon Sep 17 00:00:00 2001 From: maltegrosse Date: Wed, 1 Feb 2023 15:47:38 +0100 Subject: [PATCH] tf --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index e9ab0f4..e344c8f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -100,7 +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 mamba install -y tensorflow-gpu cudatoolkit>=11.2 -c conda-forge +RUN mamba install -y tensorflow==2.11.0=cuda112py310* cudatoolkit>=11.2 -c conda-forge #RUN mamba install -y -c conda-forge tensorflow-gpu WORKDIR /tmp RUN mamba install --quiet --yes \