ZhengPeng7 commited on
Commit
66ee2ea
1 Parent(s): bcf6dd3

Upgrade torch to 2.x for the compatibility with ZeroGPU.

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- torch==1.13.1
2
- torchvision==0.14.1
3
  opencv-python
4
  fvcore
 
1
+ torch==2.0.1
2
+ torchvision==0.15.2
3
  opencv-python
4
  fvcore