Update build-oneflow.yml

Oneflow-Inc · Nov 14, 2024 · 8d0ced2 · 8d0ced2
1 parent f644f8e
commit 8d0ced2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-oneflow.yml b/.github/workflows/build-oneflow.yml
@@ -37,11 +37,11 @@ jobs:
           - tag-suffix: "cuda12.6"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "12.6.2-cudnn-devel-rockylinux8"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.6.2-cudnn-devel-rockylinux8"
           - tag-suffix: "cuda12.5"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "12.5.2-cudnn-devel-rockylinux8"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.5.2-cudnn-devel-rockylinux8"
           - tag-suffix: "cuda12.4"
             policy: "manylinux2014"
             platform: "x86_64"