[MISC] Upgrade dependency to PyTorch 2.3.1 (#5327)

This commit is contained in:
Cody Yu
2024-07-12 12:04:26 -07:00
committed by GitHub
parent 4dbebd03cc
commit aa48e502fb
5 changed files with 8 additions and 8 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.3.0'] # Must be the most recent version that meets requirements-cuda.txt.
pytorch-version: ['2.3.1'] # Must be the most recent version that meets requirements-cuda.txt.
cuda-version: ['11.8', '12.1']
steps: