From 1791d87b6f83174c80c580abf287bc31f843168a Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 24 Feb 2026 22:35:25 -0500 Subject: [PATCH] build axolotl images with torch 2.10.0 (#3430) --- .github/workflows/main.yml | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index e081f2127..b8ebe2aeb 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -34,6 +34,12 @@ jobs: axolotl_extras: platforms: "linux/amd64,linux/arm64" is_latest: true + - cuda: 128 + cuda_version: 12.8.1 + python_version: "3.12" + pytorch: 2.10.0 + axolotl_extras: + platforms: "linux/amd64,linux/arm64" - cuda: 129 cuda_version: 12.9.1 python_version: "3.12" @@ -46,6 +52,12 @@ jobs: pytorch: 2.9.1 axolotl_extras: platforms: "linux/amd64,linux/arm64" + - cuda: 130 + cuda_version: 13.0.0 + python_version: "3.12" + pytorch: 2.10.0 + axolotl_extras: + platforms: "linux/amd64,linux/arm64" runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -112,6 +124,12 @@ jobs: axolotl_extras: is_latest: true platforms: "linux/amd64,linux/arm64" + - cuda: 128 + cuda_version: 12.8.1 + python_version: "3.12" + pytorch: 2.10.0 + axolotl_extras: + platforms: "linux/amd64,linux/arm64" - cuda: 129 cuda_version: 12.9.1 python_version: "3.12" @@ -124,6 +142,12 @@ jobs: pytorch: 2.9.1 axolotl_extras: platforms: "linux/amd64,linux/arm64" + - cuda: 130 + cuda_version: 13.0.0 + python_version: "3.12" + pytorch: 2.10.0 + axolotl_extras: + platforms: "linux/amd64,linux/arm64" runs-on: axolotl-gpu-runner steps: - name: Checkout