diff --git a/.drone.yml b/.drone.yml
index 4e8689a..a953d5e 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -4,12 +4,10 @@ name: gpu-test
 
 steps:
   - name: gpu-check
-    image: nvidia/cuda:12.0.0-base-ubuntu20.04 
+    image: nvidia/cuda:12.1.0-base-ubuntu22.04
+    pull: if-not-exists
     commands:
       - nvidia-smi
-    resources:
-      requests:
-        gpu: 1
 
 # ---
 # kind: pipeline