add

Oneflow-Inc · Oct 25, 2023 · fb06938 · fb06938
1 parent 6078574
commit fb06938
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -34,6 +34,18 @@ jobs:
       matrix:
         tag-suffix: ["cuda11.2", "cuda10.2"]
         include:
+          - tag-suffix: "cuda12.2"
+            policy: "manylinux2014"
+            platform: "x86_64"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.2.2-cudnn8-devel-centos7"
+          - tag-suffix: "cuda12.1"
+            policy: "manylinux2014"
+            platform: "x86_64"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.1.1-cudnn8-devel-centos7"
+          - tag-suffix: "cuda12.0"
+            policy: "manylinux2014"
+            platform: "x86_64"
+            CUDA_BASE_IMAGE: "nvidia/cuda:12.0.1-cudnn8-devel-centos7"
           - tag-suffix: "cuda11.8"
             policy: "manylinux2014"
             platform: "x86_64"