diff --git a/.drone.yml b/.drone.yml
index e5551df..c7bcae6 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -7,7 +7,7 @@ concurrency:
 
 steps:
   - name: install
-    image: nvcr.io/nvidia/pytorch:23.03-py3
+    image: pytorch/pytorch:latest
     privileged: true
     environment:
       mount_command: