From c5587b45accdc50e0be7b2ec9ed3a66879d68156 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 30 Jul 2024 08:50:23 -0400 Subject: [PATCH] use 12.4.1 instead of 12.4 [skip-ci] (#1796) --- .github/workflows/base.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 3a0c143df..9101fc2be 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -25,7 +25,7 @@ jobs: 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 + cuda_version: 12.4.1 cudnn_version: "" python_version: "3.11" pytorch: 2.4.0