diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index ea1da6684..2e8db61cc 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -32,6 +32,11 @@ jobs: python_version: "3.11" pytorch: 2.2.1 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" + - cuda: "121" + cuda_version: 12.1.0 + python_version: "3.11" + pytorch: 2.3.0 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" steps: - name: Checkout uses: actions/checkout@v3 diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 07a271769..f51c42a82 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -30,6 +30,11 @@ jobs: python_version: "3.11" pytorch: 2.2.1 axolotl_extras: + - cuda: 121 + cuda_version: 12.1.0 + python_version: "3.11" + pytorch: 2.3.0 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -86,6 +91,11 @@ jobs: python_version: "3.11" pytorch: 2.2.1 axolotl_extras: + - cuda: 121 + cuda_version: 12.1.0 + python_version: "3.11" + pytorch: 2.3.0 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout diff --git a/.github/workflows/nightlies.yml b/.github/workflows/nightlies.yml index cc263a887..f4172fa40 100644 --- a/.github/workflows/nightlies.yml +++ b/.github/workflows/nightlies.yml @@ -29,6 +29,11 @@ jobs: python_version: "3.11" pytorch: 2.2.1 axolotl_extras: + - cuda: 121 + cuda_version: 12.1.0 + python_version: "3.11" + pytorch: 2.3.0 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -86,6 +91,11 @@ jobs: python_version: "3.11" pytorch: 2.2.1 axolotl_extras: + - cuda: 121 + cuda_version: 12.1.0 + python_version: "3.11" + pytorch: 2.3.0 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout