deprecate: pytorch 2.0.1 image (#1315) [skip ci]
Browse files* deprecate: pytorch 2.0.1 image
* deprecate from main image
* Update main.yml
* Update tests.yml
- .github/workflows/base.yml +0 -5
- .github/workflows/main.yml +0 -10
- .github/workflows/tests.yml +1 -1
.github/workflows/base.yml
CHANGED
|
@@ -12,11 +12,6 @@ jobs:
|
|
| 12 |
fail-fast: false
|
| 13 |
matrix:
|
| 14 |
include:
|
| 15 |
-
- cuda: "118"
|
| 16 |
-
cuda_version: 11.8.0
|
| 17 |
-
python_version: "3.10"
|
| 18 |
-
pytorch: 2.0.1
|
| 19 |
-
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
|
| 20 |
- cuda: "118"
|
| 21 |
cuda_version: 11.8.0
|
| 22 |
python_version: "3.10"
|
|
|
|
| 12 |
fail-fast: false
|
| 13 |
matrix:
|
| 14 |
include:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
- cuda: "118"
|
| 16 |
cuda_version: 11.8.0
|
| 17 |
python_version: "3.10"
|
.github/workflows/main.yml
CHANGED
|
@@ -13,11 +13,6 @@ jobs:
|
|
| 13 |
fail-fast: false
|
| 14 |
matrix:
|
| 15 |
include:
|
| 16 |
-
- cuda: 118
|
| 17 |
-
cuda_version: 11.8.0
|
| 18 |
-
python_version: "3.10"
|
| 19 |
-
pytorch: 2.0.1
|
| 20 |
-
axolotl_extras:
|
| 21 |
- cuda: 118
|
| 22 |
cuda_version: 11.8.0
|
| 23 |
python_version: "3.10"
|
|
@@ -73,11 +68,6 @@ jobs:
|
|
| 73 |
strategy:
|
| 74 |
matrix:
|
| 75 |
include:
|
| 76 |
-
- cuda: 118
|
| 77 |
-
cuda_version: 11.8.0
|
| 78 |
-
python_version: "3.10"
|
| 79 |
-
pytorch: 2.0.1
|
| 80 |
-
axolotl_extras:
|
| 81 |
- cuda: 118
|
| 82 |
cuda_version: 11.8.0
|
| 83 |
python_version: "3.10"
|
|
|
|
| 13 |
fail-fast: false
|
| 14 |
matrix:
|
| 15 |
include:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
- cuda: 118
|
| 17 |
cuda_version: 11.8.0
|
| 18 |
python_version: "3.10"
|
|
|
|
| 68 |
strategy:
|
| 69 |
matrix:
|
| 70 |
include:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
- cuda: 118
|
| 72 |
cuda_version: 11.8.0
|
| 73 |
python_version: "3.10"
|
.github/workflows/tests.yml
CHANGED
|
@@ -69,7 +69,7 @@ jobs:
|
|
| 69 |
- cuda: 118
|
| 70 |
cuda_version: 11.8.0
|
| 71 |
python_version: "3.10"
|
| 72 |
-
pytorch: 2.
|
| 73 |
- cuda: 121
|
| 74 |
cuda_version: 12.1.0
|
| 75 |
python_version: "3.10"
|
|
|
|
| 69 |
- cuda: 118
|
| 70 |
cuda_version: 11.8.0
|
| 71 |
python_version: "3.10"
|
| 72 |
+
pytorch: 2.1.2
|
| 73 |
- cuda: 121
|
| 74 |
cuda_version: 12.1.0
|
| 75 |
python_version: "3.10"
|