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
|