diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 5032328..fadbebd 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -15,7 +15,7 @@ jobs:
   build_release:
     runs-on: [ubuntu-latest]
     container:
-      image: nvidia/cuda:11.8.0-devel-ubuntu22.04
+      image: nvidia/cuda:12.6.1-devel-ubuntu22.04
     permissions:
       contents: write