Unverified Commit c8357f4e authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

Setup: Revision - Revise torch extra_require (#177)

**Description**
change the minimal version requirement for superbench:
```
'torch>=1.7.0a0',
'torchvision>=0.8.0a0',
```
parent 8cd264fd
......@@ -159,8 +159,8 @@ def run(self):
'yapf>=0.30.0',
],
'torch': [
'torch>=1.7.0',
'torchvision>=0.8.0',
'torch>=1.7.0a0',
'torchvision>=0.8.0a0',
'transformers>=4.3.3',
],
'nvidia': ['py3nvml>=0.2.6']
......
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