From d8d1788ffc4dd1d58bd5813a83abf6f1f8fad60f Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 30 Jul 2024 08:06:11 -0400 Subject: [PATCH] move to supporting mostly 12.1 w 2.3.1 and add new 12.4 with 2.4.0 (#1793) --- .github/workflows/base.yml | 33 +++++++++------------------------ 1 file changed, 9 insertions(+), 24 deletions(-) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index f8eaff270..4019a5baf 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -12,36 +12,21 @@ jobs: fail-fast: false matrix: include: - - cuda: "118" - cuda_version: 11.8.0 + - cuda: "121" + cuda_version: 12.1.1 python_version: "3.10" - pytorch: 2.1.2 + pytorch: 2.3.1 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - cuda: "121" - cuda_version: 12.1.0 - python_version: "3.10" - pytorch: 2.1.2 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - - cuda: "121" - cuda_version: 12.1.0 - python_version: "3.11" - pytorch: 2.1.2 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - - cuda: "121" - cuda_version: 12.1.0 - python_version: "3.11" - pytorch: 2.2.2 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - - cuda: "121" - cuda_version: 12.1.0 - python_version: "3.11" - pytorch: 2.3.0 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - - cuda: "121" - cuda_version: 12.1.0 + cuda_version: 12.1.1 python_version: "3.11" pytorch: 2.3.1 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + - cuda: "124" + cuda_version: 12.4.0 + python_version: "3.11" + pytorch: 2.4.0 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" steps: - name: Checkout uses: actions/checkout@v3