RockeyCoss commited on
Commit
b663f97
·
1 Parent(s): 51f6859

modify requirement

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==1.10.0
2
- torchvision==0.11.0
3
  cython
4
  numpy
5
  matplotlib
 
1
+ torch==1.10.0+cpu
2
+ torchvision==0.11.0+cpu
3
  cython
4
  numpy
5
  matplotlib