Merge pull request #24 from coreweave/es/torch-cu121

build(torch): Add CUDA 12.1 builds
coreweave · Jun 27, 2023 · a0333d3 · a0333d3
2 parents 209c517 + 06d849f
commit a0333d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/torch-base.yml b/.github/workflows/torch-base.yml
@@ -12,7 +12,7 @@ jobs:
   build:
     strategy:
       matrix:
-        cuda: [12.0.1, 11.8.0]
+        cuda: [12.1.1, 12.0.1, 11.8.0]
         include:
           - torch: 2.0.1
             vision: 0.15.2

diff --git a/.github/workflows/torch-nccl.yml b/.github/workflows/torch-nccl.yml
@@ -13,6 +13,9 @@ jobs:
     strategy:
       matrix:
         image:
+          - cuda: 12.1.1
+            nccl: 2.18.1-1
+            nccl-tests-hash: e90a40f
           - cuda: 12.0.1
             nccl: 2.18.1-1
             nccl-tests-hash: e90a40f