Hanxiao Xiang commited on
Commit
8326b90
1 Parent(s): da8b85a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -3,5 +3,5 @@ openmim==0.3.7
3
  pillow==9.4.0
4
  pycocotools==2.0.7
5
  scipy==1.10.1
6
- torch==2.0.0
7
  torchvision==0.15.0
 
3
  pillow==9.4.0
4
  pycocotools==2.0.7
5
  scipy==1.10.1
6
+ torch==2.0.0+cu117
7
  torchvision==0.15.0