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

---
 .woodpecker.yml | 1 +
 Dockerfile      | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.woodpecker.yml b/.woodpecker.yml
index 7e1d384..c00ebed 100644
--- a/.woodpecker.yml
+++ b/.woodpecker.yml
@@ -5,6 +5,7 @@ pipeline:
     group: docker
     environment:
      - CONDA_CUDA_OVERRIDE=11.2
+     - NVIDIA_VISIBLE_DEVICES=all
     settings:
       registry: https://git.sandbox.iuk.hdm-stuttgart.de
       repo: git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience-gpu
diff --git a/Dockerfile b/Dockerfile
index e344c8f..d4cbd4f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -100,7 +100,8 @@ 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==2.11.0=cuda112py310* cudatoolkit>=11.2 -c conda-forge
+RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
+#==2.11.0=cuda*
 #RUN mamba install -y -c conda-forge tensorflow-gpu
 WORKDIR /tmp
 RUN mamba install --quiet --yes \