Update requirements.txt
Browse files- 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
|