diff --git a/.github/workflows/build-docker.yml b/.github/workflows/build_docker.yml
similarity index 71%
rename from .github/workflows/build-docker.yml
rename to .github/workflows/build_docker.yml
index 896d3bb..fd59a9f 100644
--- a/.github/workflows/build-docker.yml
+++ b/.github/workflows/build_docker.yml
@@ -4,14 +4,11 @@ on:
 
 jobs:
   publish_images:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04
+    container: nvidia/cuda:12.6.3-devel-ubuntu24.04
     steps:
-      - uses: Jimver/cuda-toolkit@v0.2.16
-        id: cuda-toolkit
-        with:
-            cuda: '12.5.0'
       - name: checkout
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
       - name: build image
         run: |
           docker build . -t aghaeifar/spinwalk:latest