diff --git a/.github/workflows/build-oneflow.yml b/.github/workflows/build-oneflow.yml
index a36b8a16..939b9d13 100644
--- a/.github/workflows/build-oneflow.yml
+++ b/.github/workflows/build-oneflow.yml
@@ -34,18 +34,18 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - tag-suffix: "cuda12.2"
+          - tag-suffix: "cuda12.6"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:12.2.2-cudnn8-devel-centos7"
-          - tag-suffix: "cuda12.1"
+            CUDA_BASE_IMAGE: "12.6.2-cudnn-devel-rockylinux8"
+          - tag-suffix: "cuda12.5"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:12.1.1-cudnn8-devel-centos7"
-          - tag-suffix: "cuda11.8"
+            CUDA_BASE_IMAGE: "12.5.2-cudnn-devel-rockylinux8"
+          - tag-suffix: "cuda12.4"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:11.8.0-cudnn8-devel-centos7"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.4.1-cudnn-devel-rockylinux8"
           - tag-suffix: "cpu"
             policy: "manylinux2014"
             platform: "x86_64"