Add CI for pytorch 1.10 (#1431)

pull/1444/head
Wenwei Zhang 2021-10-24 13:32:31 +08:00 committed by GitHub
parent 5b5b47fc87
commit ef02219679
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions

View File

@ -211,10 +211,12 @@ jobs:
strategy:
matrix:
python-version: [3.7]
torch: [1.9.0+cu102]
torch: [1.9.0+cu102, 1.10.0+cu102]
include:
- torch: 1.9.0+cu102
torchvision: 0.10.0+cu102
- torch: 1.10.0+cu102
torchvision: 0.11.0+cu102
- python-version: 3.6
torch: 1.9.0+cu102
torchvision: 0.10.0+cu102