diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 10e274c..88c1fc7 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -64,6 +64,9 @@ jobs: cuda-version: "12.6.3" - torch-version: "2.0.1" cuda-version: "12.8.1" + # torch 2.6.0 does not support CUDA 12.1 + - torch-version: "2.6.0" + cuda-version: "12.1.1" # torch 2.7.0 does not support CUDA 12.4 - torch-version: "2.7.0" cuda-version: "12.4.1" @@ -178,6 +181,9 @@ jobs: cuda-version: "12.6.3" - torch-version: "2.0.1" cuda-version: "12.8.1" + # torch 2.6.0 does not support CUDA 12.1 + - torch-version: "2.6.0" + cuda-version: "12.1.1" # torch 2.7.0 does not support CUDA 12.4 - torch-version: "2.7.0" cuda-version: "12.4.1"