From d0bb48898e8edb529b45f99223f019f0a639249f Mon Sep 17 00:00:00 2001
From: maltegrosse <malte.grosse@googlemail.com>
Date: Wed, 1 Feb 2023 16:43:24 +0100
Subject: [PATCH] gpu

---
 Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 4003f79..e7b041c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -100,10 +100,11 @@ 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 nvidia-smi
+ENV NUM_GPUS=1
 #RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
 #==2.11.0=cuda*
-RUN mamba install -y -c conda-forge tensorflow-gpu
+#RUN mamba install -y -c conda-forge tensorflow-gpu
+RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge
 WORKDIR /tmp
 RUN mamba install --quiet --yes \
     'notebook' \