tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

add base docker image with pytorch 2.7.0 and variant for cuda 12.8 (#2551)

Browse Source 
* add base docker image with pytorch 2.7.0 and variant for cuda 12.8

* my bash is terrible
This commit is contained in:
Wing Lian
2025-04-23 14:59:03 -04:00
committed by GitHub
parent c4053481ff
commit 0d691cc2a7
3 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
.github/workflows/base.yml vendored
View File

@@ -46,6 +46,18 @@ jobs:
python_version: "3.11"
pytorch: 2.6.0
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "126"
cuda_version: 12.6.3
cudnn_version: ""
python_version: "3.11"
pytorch: 2.7.0
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "128"
cuda_version: 12.6.3
cudnn_version: ""
python_version: "3.11"
pytorch: 2.7.0
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "128"
cuda_version: 12.8.1
cudnn_version: ""