Skip to content

Commit 383f220

Browse files
authored
build torch 2.9.0 base images (#3221)
1 parent 8bb871b commit 383f220

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

.github/workflows/base.yml

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,13 @@ jobs:
5353
pytorch: 2.8.0
5454
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
5555
dockerfile: "Dockerfile-base"
56+
- cuda: "128"
57+
cuda_version: 12.8.1
58+
cudnn_version: ""
59+
python_version: "3.11"
60+
pytorch: 2.9.0
61+
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
62+
dockerfile: "Dockerfile-base"
5663
# - cuda: "128"
5764
# cuda_version: 12.8.1
5865
# cudnn_version: ""
@@ -129,6 +136,13 @@ jobs:
129136
pytorch: 2.8.0
130137
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
131138
dockerfile: "Dockerfile-uv-base"
139+
- cuda: "128"
140+
cuda_version: 12.8.1
141+
cudnn_version: ""
142+
python_version: "3.11"
143+
pytorch: 2.9.0
144+
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
145+
dockerfile: "Dockerfile-uv-base"
132146
steps:
133147
- name: Checkout
134148
uses: actions/checkout@v4

0 commit comments

Comments
 (0)