diff --git a/.woodpecker.yml b/.woodpecker.yml
index 2fcb81c..498fe07 100644
--- a/.woodpecker.yml
+++ b/.woodpecker.yml
@@ -7,7 +7,7 @@ steps:
     settings:
       registry: git.sandbox.iuk.hdm-stuttgart.de
       repository: grosse/jupyterlab-datascience-gpu
-      tag: 4.0.x-gpu-torch
+      tag: 4.0.x-gpu-tf
       architectures: amd64
       context: Dockerfile
       loglevel: debug
diff --git a/Dockerfile b/Dockerfile
index 3252107..124fa50 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,4 +5,4 @@ ARG NB_UID="1000"
 ARG NB_GID="100"
 ENV CONDA_CUDA_OVERRIDE="11.2"  
 ENV NUM_GPUS=1
-RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
\ No newline at end of file
+RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge
\ No newline at end of file