Skip to content

GitLab

Unverified Commit ef022196 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

Add CI for pytorch 1.10 (#1431)

parent 5b5b47fc
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
.github/workflows/build.yml
View file @ ef022196
...@@ -211,10 +211,12 @@ jobs: ...@@ -211,10 +211,12 @@ jobs:
strategy: strategy:
matrix: matrix:
python-version: [3.7] python-version: [3.7]
torch: [1.9.0+cu102] torch: [1.9.0+cu102, 1.10.0+cu102]
include: include:
- torch: 1.9.0+cu102 - torch: 1.9.0+cu102
torchvision: 0.10.0+cu102 torchvision: 0.10.0+cu102
- torch: 1.10.0+cu102
torchvision: 0.11.0+cu102
- python-version: 3.6 - python-version: 3.6
torch: 1.9.0+cu102 torch: 1.9.0+cu102
torchvision: 0.10.0+cu102 torchvision: 0.10.0+cu102
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment