From c5e5aba54755cb1c827004b8ef404bd9b3d3ee4c Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 8 Aug 2025 02:30:16 -0400 Subject: [PATCH] Add 2.8.0 base images and uv images (#3034) --- .github/workflows/base.yml | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index e97e73442..160ed7df9 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -54,7 +54,7 @@ jobs: torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" dockerfile: "Dockerfile-base" - cuda: "128" - cuda_version: 12.6.3 + cuda_version: 12.8.1 cudnn_version: "" python_version: "3.11" pytorch: 2.7.1 @@ -64,9 +64,16 @@ jobs: cuda_version: 12.8.1 cudnn_version: "" python_version: "3.11" - pytorch: nightly + pytorch: 2.8.0 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" - dockerfile: "Dockerfile-base-nightly" + dockerfile: "Dockerfile-base" +# - cuda: "128" +# cuda_version: 12.8.1 +# cudnn_version: "" +# python_version: "3.11" +# pytorch: nightly +# torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" +# dockerfile: "Dockerfile-base-nightly" # # "next" is for release candidates of pytorch # - cuda: "128" # cuda_version: 12.8.1 @@ -122,6 +129,13 @@ jobs: pytorch: 2.6.0 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" dockerfile: "Dockerfile-uv-base" + - cuda: "126" + cuda_version: 12.6.3 + cudnn_version: "" + python_version: "3.11" + pytorch: 2.7.1 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + dockerfile: "Dockerfile-uv-base" - cuda: "128" cuda_version: 12.8.1 cudnn_version: "" @@ -129,6 +143,13 @@ jobs: pytorch: 2.7.1 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" dockerfile: "Dockerfile-uv-base" + - cuda: "128" + cuda_version: 12.8.1 + cudnn_version: "" + python_version: "3.11" + pytorch: 2.8.0 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + dockerfile: "Dockerfile-uv-base" steps: - name: Checkout uses: actions/checkout@v4