support for cuda 12.1 (#989)
Browse files- .github/workflows/base.yml +5 -0
- .github/workflows/main.yml +10 -0
.github/workflows/base.yml
CHANGED
|
@@ -30,6 +30,11 @@ jobs:
|
|
| 30 |
python_version: "3.10"
|
| 31 |
pytorch: 2.1.1
|
| 32 |
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
steps:
|
| 34 |
- name: Checkout
|
| 35 |
uses: actions/checkout@v3
|
|
|
|
| 30 |
python_version: "3.10"
|
| 31 |
pytorch: 2.1.1
|
| 32 |
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
|
| 33 |
+
- cuda: "121"
|
| 34 |
+
cuda_version: 12.1.0
|
| 35 |
+
python_version: "3.10"
|
| 36 |
+
pytorch: 2.1.1
|
| 37 |
+
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
|
| 38 |
steps:
|
| 39 |
- name: Checkout
|
| 40 |
uses: actions/checkout@v3
|
.github/workflows/main.yml
CHANGED
|
@@ -29,6 +29,11 @@ jobs:
|
|
| 29 |
python_version: "3.10"
|
| 30 |
pytorch: 2.1.1
|
| 31 |
axolotl_extras:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
runs-on: [self-hosted, gpu, docker]
|
| 33 |
steps:
|
| 34 |
- name: Checkout
|
|
@@ -95,6 +100,11 @@ jobs:
|
|
| 95 |
python_version: "3.10"
|
| 96 |
pytorch: 2.1.1
|
| 97 |
axolotl_extras:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 98 |
runs-on: [self-hosted, gpu, docker]
|
| 99 |
steps:
|
| 100 |
- name: Checkout
|
|
|
|
| 29 |
python_version: "3.10"
|
| 30 |
pytorch: 2.1.1
|
| 31 |
axolotl_extras:
|
| 32 |
+
- cuda: 121
|
| 33 |
+
cuda_version: 12.1.0
|
| 34 |
+
python_version: "3.10"
|
| 35 |
+
pytorch: 2.1.1
|
| 36 |
+
axolotl_extras:
|
| 37 |
runs-on: [self-hosted, gpu, docker]
|
| 38 |
steps:
|
| 39 |
- name: Checkout
|
|
|
|
| 100 |
python_version: "3.10"
|
| 101 |
pytorch: 2.1.1
|
| 102 |
axolotl_extras:
|
| 103 |
+
- cuda: 121
|
| 104 |
+
cuda_version: 12.1.0
|
| 105 |
+
python_version: "3.10"
|
| 106 |
+
pytorch: 2.1.1
|
| 107 |
+
axolotl_extras:
|
| 108 |
runs-on: [self-hosted, gpu, docker]
|
| 109 |
steps:
|
| 110 |
- name: Checkout
|