diff --git a/.woodpecker.yml b/.woodpecker.yml
index 1b69c34..7ac853d 100644
--- a/.woodpecker.yml
+++ b/.woodpecker.yml
@@ -5,7 +5,7 @@ pipeline:
     commands:
       - echo "pick pick wood pickasdpaasdsdick ....asd.sd..kj......"
   nvidia-test:
-    image: nvidia/cuda
+    image: nvidia/cuda:11.6.2-base-ubuntu20.04
     commands:
       - nvidia-smi
       - sleep 30