use the same version pinning as PyPI
@@ -22,8 +22,8 @@ requirements:
run:
- python >=3.6
- opencv >=4.1
- - pytorch >=1.7.0
- - torchvision >=0.8.2
+ - pytorch ==1.7.0
+ - torchvision ==0.8.2
- tqdm >=4.42
- ttach >=0.0.3
- typing-extensions ==3.7.4.3
