[CI/CD] Upgrade PyTorch version to v2.1.1 (#2045)

This commit is contained in:
Woosuk Kwon
2023-12-11 17:48:11 -08:00
committed by GitHub
parent 31d2ab4aff
commit f3e024bece
3 changed files with 4 additions and 4 deletions

View File

@@ -49,7 +49,7 @@ jobs:
matrix:
os: ['ubuntu-20.04']
python-version: ['3.8', '3.9', '3.10', '3.11']
pytorch-version: ['2.1.0']
pytorch-version: ['2.1.1']
cuda-version: ['11.8', '12.1']
steps: