ensure the tests use the same version of torch as the latest base docker images (#1215) [skip ci]

This commit is contained in:
Wing Lian
2024-01-26 10:38:30 -05:00
committed by GitHub
parent 62ca4a2b71
commit 1b180034c7

View File

@@ -73,7 +73,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.10"
pytorch: 2.1.1
pytorch: 2.1.2
steps:
- name: Checkout
uses: actions/checkout@v4