diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 3a0c143df5..9101fc2bea 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