use torch img

Author: jackalcooper

.github/workflows/build-oneflow.yml

@@ -37,19 +37,19 @@ jobs:
           - tag-suffix: "cuda12.4"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:12.4.1-cudnn-devel-rockylinux8"
+            CUDA_BASE_IMAGE: "pytorch/manylinux-builder:cuda12.4-main"
           - tag-suffix: "cuda12.2"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:12.2.2-cudnn8-devel-centos7"
+            CUDA_BASE_IMAGE: "pytorch/manylinux-builder:cuda12.2-main"
           - tag-suffix: "cuda12.1"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:12.1.1-cudnn8-devel-centos7"
+            CUDA_BASE_IMAGE: "pytorch/manylinux-builder:cuda12.1-main"
           - tag-suffix: "cuda11.8"
             policy: "manylinux2014"
             platform: "x86_64"
-            CUDA_BASE_IMAGE: "nvidia/cuda:11.8.0-cudnn8-devel-centos7"
+            CUDA_BASE_IMAGE: "pytorch/manylinux-builder:cuda11.8-main7"
           - tag-suffix: "cpu"
             policy: "manylinux2014"
             platform: "x86_64"