stevengrove commited on
Commit
896835c
1 Parent(s): db858ba

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,5 +1,3 @@
1
- torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
2
- torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
3
  openmim
4
  mmcv>=2.0.0rc4,<2.1.0
5
  mmdet>=3.0.0
@@ -22,3 +20,7 @@ onnxruntime
22
  onnxruntime-gpu
23
  onnx-simplifier
24
 
 
 
 
 
 
 
 
1
  openmim
2
  mmcv>=2.0.0rc4,<2.1.0
3
  mmdet>=3.0.0
 
20
  onnxruntime-gpu
21
  onnx-simplifier
22
 
23
+ --index-url https://download.pytorch.org/whl/cu118
24
+ torch==2.0.1
25
+ --index-url https://download.pytorch.org/whl/cu118
26
+ torchvision==0.15.2