larger torch cuda arch list coverage for older cuda

add pytorch 2.11 to base images
2026-04-23 15:05:01 -04:00 · 2026-04-23 14:57:49 -04:00
1 changed files with 16 additions and 0 deletions
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -224,6 +224,22 @@ jobs:
            torch_cuda_arch_list: "9.0+PTX"
            dockerfile: "Dockerfile-uv-base"
            platforms: "linux/amd64,linux/arm64"
+          - cuda: "128"
+            cuda_version: 12.8.1
+            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"
+            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
+            torch_cuda_arch_list: "9.0+PTX"
+            dockerfile: "Dockerfile-uv-base"
+            platforms: "linux/amd64,linux/arm64"
    steps:
      - name: Checkout
        uses: actions/checkout@v4
Author	SHA1	Message	Date
Wing Lian	98e18d59d2	larger torch cuda arch list coverage for older cuda	2026-04-23 15:05:01 -04:00
Wing Lian	462135acfb	add pytorch 2.11 to base images	2026-04-23 14:57:49 -04:00