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

add torch 2.3.0 to builds (#1593)

Browse Source
This commit is contained in:
Wing Lian
2024-05-05 18:45:45 -04:00
committed by GitHub
parent 120b809465
commit 70185763f6
3 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,6 +32,11 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.2.1 pytorch: 2.2.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" 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: steps:
- name: Checkout - name: Checkout
uses: actions/checkout@v3 uses: actions/checkout@v3

10
.github/workflows/main.yml vendored
View File

@@ -30,6 +30,11 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.2.1 pytorch: 2.2.1
axolotl_extras: axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.3.0
axolotl_extras:
runs-on: axolotl-gpu-runner runs-on: axolotl-gpu-runner
steps: steps:
- name: Checkout - name: Checkout
@@ -86,6 +91,11 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.2.1 pytorch: 2.2.1
axolotl_extras: axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.3.0
axolotl_extras:
runs-on: axolotl-gpu-runner runs-on: axolotl-gpu-runner
steps: steps:
- name: Checkout - name: Checkout

10
.github/workflows/nightlies.yml vendored
View File

@@ -29,6 +29,11 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.2.1 pytorch: 2.2.1
axolotl_extras: axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.3.0
axolotl_extras:
runs-on: axolotl-gpu-runner runs-on: axolotl-gpu-runner
steps: steps:
- name: Checkout - name: Checkout
@@ -86,6 +91,11 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.2.1 pytorch: 2.2.1
axolotl_extras: axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.3.0
axolotl_extras:
runs-on: axolotl-gpu-runner runs-on: axolotl-gpu-runner
steps: steps:
- name: Checkout - name: Checkout