ymcmy commited on
Commit
681ca6a
·
verified ·
1 Parent(s): 7bcbdef

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  numpy==1.24.0
2
- torch==2.0.0
3
- torchvision==0.15.0
4
  opencv-python==4.5.5.62
5
  Pillow==9.4.0
6
  scipy==1.10.0
 
1
  numpy==1.24.0
2
+ torch==2.0.0+cu117
3
+ torchvision==0.15.0+cu117
4
  opencv-python==4.5.5.62
5
  Pillow==9.4.0
6
  scipy==1.10.0