diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 90dd2b3db0..ed987558fe 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -38,14 +38,6 @@ jobs: torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" dockerfile: "Dockerfile-base" platforms: "linux/amd64,linux/arm64" - - cuda: "128" - cuda_version: 12.8.1 - cudnn_version: "" - python_version: "3.11" - pytorch: 2.10.0 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - dockerfile: "Dockerfile-base" - platforms: "linux/amd64,linux/arm64" - cuda: "128" cuda_version: 12.8.1 cudnn_version: "" @@ -70,14 +62,6 @@ jobs: torch_cuda_arch_list: "9.0 10.0 10.3 12.0+PTX" dockerfile: "Dockerfile-base" platforms: "linux/amd64,linux/arm64" - - cuda: "130" - cuda_version: 13.0.0 - cudnn_version: "" - python_version: "3.12" - pytorch: 2.9.1 - torch_cuda_arch_list: "9.0 10.0 10.3 12.0+PTX" - dockerfile: "Dockerfile-base" - platforms: "linux/amd64,linux/arm64" - cuda: "130" cuda_version: 13.0.0 cudnn_version: "" @@ -208,19 +192,19 @@ jobs: torch_cuda_arch_list: "9.0 10.0 10.3 12.0+PTX" dockerfile: "Dockerfile-uv-base" platforms: "linux/amd64,linux/arm64" - - cuda: "128" - cuda_version: 12.8.1 + - cuda: "130" + cuda_version: 13.0.0 cudnn_version: "" python_version: "3.12" pytorch: 2.11.0 - torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + torch_cuda_arch_list: "9.0 10.0 10.3 12.0+PTX" dockerfile: "Dockerfile-uv-base" platforms: "linux/amd64,linux/arm64" - cuda: "130" cuda_version: 13.0.0 cudnn_version: "" python_version: "3.12" - pytorch: 2.11.0 + pytorch: 2.12.0 torch_cuda_arch_list: "9.0 10.0 10.3 12.0+PTX" dockerfile: "Dockerfile-uv-base" platforms: "linux/amd64,linux/arm64" diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 8c97f8b40e..1adc76bf61 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -68,7 +68,7 @@ jobs: fail-fast: false matrix: python_version: ["3.12", "3.14"] - pytorch_version: ["2.9.1", "2.10.0"] + pytorch_version: ["2.9.1", "2.10.0", "2.11.0", "2.12.0"] exclude: - python_version: "3.14" pytorch_version: "2.9.1" @@ -155,7 +155,7 @@ jobs: fail-fast: false matrix: python_version: ["3.12", "3.14"] - pytorch_version: ["2.9.1", "2.10.0"] + pytorch_version: ["2.9.1", "2.10.0", "2.11.0", "2.12.0"] exclude: - python_version: "3.14" pytorch_version: "2.9.1"