diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 47d1a89..d0164f9 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -21,6 +21,8 @@ jobs:
     permissions:
       id-token: write
       contents: write
+    container:
+      image: nvidia/cuda:12.4.1-devel-ubuntu22.04
 
     environment:
       name: pypi