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