ensure the tests use the same version of torch as the latest base docker images (#1215) [skip ci]
Browse files
.github/workflows/tests.yml
CHANGED
|
@@ -73,7 +73,7 @@ jobs:
|
|
| 73 |
- cuda: 121
|
| 74 |
cuda_version: 12.1.0
|
| 75 |
python_version: "3.10"
|
| 76 |
-
pytorch: 2.1.
|
| 77 |
steps:
|
| 78 |
- name: Checkout
|
| 79 |
uses: actions/checkout@v4
|
|
|
|
| 73 |
- cuda: 121
|
| 74 |
cuda_version: 12.1.0
|
| 75 |
python_version: "3.10"
|
| 76 |
+
pytorch: 2.1.2
|
| 77 |
steps:
|
| 78 |
- name: Checkout
|
| 79 |
uses: actions/checkout@v4
|