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

deprecate: pytorch 2.0.1 image (#1315) [skip ci]

Browse Source 
* deprecate: pytorch 2.0.1 image

* deprecate from main image

* Update main.yml

* Update tests.yml
This commit is contained in:
NanoCode012
2024-02-22 11:39:47 +09:00
committed by GitHub
parent 2752d5f958
commit a359579371
3 changed files with 1 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,11 +12,6 @@ jobs:
fail-fast: false
matrix:
include:
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.10"

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

@@ -13,11 +13,6 @@ jobs:
fail-fast: false
matrix:
include:
- cuda: 118
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.1
axolotl_extras:
- cuda: 118
cuda_version: 11.8.0
python_version: "3.10"
@@ -73,11 +68,6 @@ jobs:
strategy:
matrix:
include:
- cuda: 118
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.1
axolotl_extras:
- cuda: 118
cuda_version: 11.8.0
python_version: "3.10"

2
.github/workflows/tests.yml vendored
View File

@@ -69,7 +69,7 @@ jobs:
- cuda: 118
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.1
pytorch: 2.1.2
- cuda: 121
cuda_version: 12.1.0
python_version: "3.10"