update torch 2.2.1 -> 2.2.2 (#1622)
Browse files- .github/workflows/base.yml +1 -1
- .github/workflows/main.yml +2 -2
- .github/workflows/nightlies.yml +2 -2
- .github/workflows/tests.yml +1 -1
- README.md +1 -1
.github/workflows/base.yml
CHANGED
|
@@ -30,7 +30,7 @@ jobs:
|
|
| 30 |
- cuda: "121"
|
| 31 |
cuda_version: 12.1.0
|
| 32 |
python_version: "3.11"
|
| 33 |
-
pytorch: 2.2.
|
| 34 |
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
|
| 35 |
- cuda: "121"
|
| 36 |
cuda_version: 12.1.0
|
|
|
|
| 30 |
- cuda: "121"
|
| 31 |
cuda_version: 12.1.0
|
| 32 |
python_version: "3.11"
|
| 33 |
+
pytorch: 2.2.2
|
| 34 |
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
|
| 35 |
- cuda: "121"
|
| 36 |
cuda_version: 12.1.0
|
.github/workflows/main.yml
CHANGED
|
@@ -28,7 +28,7 @@ jobs:
|
|
| 28 |
- cuda: 121
|
| 29 |
cuda_version: 12.1.0
|
| 30 |
python_version: "3.11"
|
| 31 |
-
pytorch: 2.2.
|
| 32 |
axolotl_extras:
|
| 33 |
- cuda: 121
|
| 34 |
cuda_version: 12.1.0
|
|
@@ -89,7 +89,7 @@ jobs:
|
|
| 89 |
- cuda: 121
|
| 90 |
cuda_version: 12.1.0
|
| 91 |
python_version: "3.11"
|
| 92 |
-
pytorch: 2.2.
|
| 93 |
axolotl_extras:
|
| 94 |
- cuda: 121
|
| 95 |
cuda_version: 12.1.0
|
|
|
|
| 28 |
- cuda: 121
|
| 29 |
cuda_version: 12.1.0
|
| 30 |
python_version: "3.11"
|
| 31 |
+
pytorch: 2.2.2
|
| 32 |
axolotl_extras:
|
| 33 |
- cuda: 121
|
| 34 |
cuda_version: 12.1.0
|
|
|
|
| 89 |
- cuda: 121
|
| 90 |
cuda_version: 12.1.0
|
| 91 |
python_version: "3.11"
|
| 92 |
+
pytorch: 2.2.2
|
| 93 |
axolotl_extras:
|
| 94 |
- cuda: 121
|
| 95 |
cuda_version: 12.1.0
|
.github/workflows/nightlies.yml
CHANGED
|
@@ -27,7 +27,7 @@ jobs:
|
|
| 27 |
- cuda: 121
|
| 28 |
cuda_version: 12.1.0
|
| 29 |
python_version: "3.11"
|
| 30 |
-
pytorch: 2.2.
|
| 31 |
axolotl_extras:
|
| 32 |
- cuda: 121
|
| 33 |
cuda_version: 12.1.0
|
|
@@ -89,7 +89,7 @@ jobs:
|
|
| 89 |
- cuda: 121
|
| 90 |
cuda_version: 12.1.0
|
| 91 |
python_version: "3.11"
|
| 92 |
-
pytorch: 2.2.
|
| 93 |
axolotl_extras:
|
| 94 |
- cuda: 121
|
| 95 |
cuda_version: 12.1.0
|
|
|
|
| 27 |
- cuda: 121
|
| 28 |
cuda_version: 12.1.0
|
| 29 |
python_version: "3.11"
|
| 30 |
+
pytorch: 2.2.2
|
| 31 |
axolotl_extras:
|
| 32 |
- cuda: 121
|
| 33 |
cuda_version: 12.1.0
|
|
|
|
| 89 |
- cuda: 121
|
| 90 |
cuda_version: 12.1.0
|
| 91 |
python_version: "3.11"
|
| 92 |
+
pytorch: 2.2.2
|
| 93 |
axolotl_extras:
|
| 94 |
- cuda: 121
|
| 95 |
cuda_version: 12.1.0
|
.github/workflows/tests.yml
CHANGED
|
@@ -82,7 +82,7 @@ jobs:
|
|
| 82 |
- cuda: 121
|
| 83 |
cuda_version: 12.1.0
|
| 84 |
python_version: "3.11"
|
| 85 |
-
pytorch: 2.2.
|
| 86 |
num_gpus: 1
|
| 87 |
steps:
|
| 88 |
- name: Checkout
|
|
|
|
| 82 |
- cuda: 121
|
| 83 |
cuda_version: 12.1.0
|
| 84 |
python_version: "3.11"
|
| 85 |
+
pytorch: 2.2.2
|
| 86 |
num_gpus: 1
|
| 87 |
steps:
|
| 88 |
- name: Checkout
|
README.md
CHANGED
|
@@ -302,7 +302,7 @@ Write a job description in YAML as below:
|
|
| 302 |
# dstack.yaml
|
| 303 |
type: task
|
| 304 |
|
| 305 |
-
image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.
|
| 306 |
|
| 307 |
env:
|
| 308 |
- HUGGING_FACE_HUB_TOKEN
|
|
|
|
| 302 |
# dstack.yaml
|
| 303 |
type: task
|
| 304 |
|
| 305 |
+
image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.2
|
| 306 |
|
| 307 |
env:
|
| 308 |
- HUGGING_FACE_HUB_TOKEN
|