diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index a28b7664a..c22dfa60d 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -52,11 +52,11 @@ jobs: python_version: "3.11" pytorch: 2.7.0 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - - cuda: "128" - cuda_version: 12.8.1 + - cuda: "126" + cuda_version: 12.6.3 cudnn_version: "" python_version: "3.11" - pytorch: 2.7.0 + pytorch: 2.6.0 suffix: "-hopper" torch_cuda_arch_list: "9.0+PTX" - cuda: "128"