From 70185763f645c704c69bd2b6fffc2d87ffb3dd73 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 5 May 2024 18:45:45 -0400 Subject: [PATCH] add torch 2.3.0 to builds (#1593) --- .github/workflows/base.yml | 5 +++++ .github/workflows/main.yml | 10 ++++++++++ .github/workflows/nightlies.yml | 10 ++++++++++ 3 files changed, 25 insertions(+) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index ea1da66840..2e8db61cc7 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 07a271769c..f51c42a823 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 cc263a887a..f4172fa406 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