Skip to content

Commit fccc712

Browse files
authored
builds for py312-cu128-torch2.9.1 (#3489)
1 parent 23ad40b commit fccc712

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

.github/workflows/base.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,14 @@ jobs:
173173
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
174174
dockerfile: "Dockerfile-uv-base"
175175
platforms: "linux/amd64,linux/arm64"
176+
- cuda: "128"
177+
cuda_version: 12.8.1
178+
cudnn_version: ""
179+
python_version: "3.12"
180+
pytorch: 2.9.1
181+
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
182+
dockerfile: "Dockerfile-uv-base"
183+
platforms: "linux/amd64,linux/arm64"
176184
- cuda: "128"
177185
cuda_version: 12.8.1
178186
cudnn_version: ""

.github/workflows/main.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,12 @@ jobs:
110110
pytorch: 2.9.1
111111
axolotl_extras:
112112
platforms: "linux/amd64,linux/arm64"
113+
- cuda: 128
114+
cuda_version: 12.8.1
115+
python_version: "3.12"
116+
pytorch: 2.9.1
117+
axolotl_extras:
118+
platforms: "linux/amd64,linux/arm64"
113119
is_latest: true
114120
- cuda: 128
115121
cuda_version: 12.8.1
@@ -193,6 +199,12 @@ jobs:
193199
python_version: "3.11"
194200
pytorch: 2.9.1
195201
axolotl_extras:
202+
platforms: "linux/amd64,linux/arm64"
203+
- cuda: 128
204+
cuda_version: 12.8.1
205+
python_version: "3.12"
206+
pytorch: 2.9.1
207+
axolotl_extras:
196208
is_latest: true
197209
platforms: "linux/amd64,linux/arm64"
198210
- cuda: 128

0 commit comments

Comments
 (0)