diff --git a/.drone.yml b/.drone.yml
index 2de9637..4e8689a 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -4,7 +4,7 @@ name: gpu-test
 
 steps:
   - name: gpu-check
-    image: nvidia/cuda:12.0-base
+    image: nvidia/cuda:12.0.0-base-ubuntu20.04 
     commands:
       - nvidia-smi
     resources: