From 5411146228ea99ede4cbe3fe0d744fdfc2398e20 Mon Sep 17 00:00:00 2001
From: maltegrosse <malte.grosse@googlemail.com>
Date: Wed, 1 Feb 2023 15:35:36 +0100
Subject: [PATCH] gpu

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 810958e..e9ab0f4 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 cudatoolkit>=11.2 -c conda-forge
+RUN mamba install -y tensorflow-gpu cudatoolkit>=11.2 -c conda-forge
 #RUN mamba install -y -c conda-forge tensorflow-gpu
 WORKDIR /tmp
 RUN mamba install --quiet --yes \