kiurtis commited on
Commit
70f4ec6
·
1 Parent(s): 13b0e50

maybe fix the gpu onnx thing

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -13,11 +13,10 @@ ultralytics==8.3.68
13
  ultralytics-thop==2.0.14
14
  #opencv-python==4.11.0.86
15
  python-dotenv==1.0.0
16
- onnxruntime==1.19.2
17
  matplotlib==3.8.1
18
  onnx==1.17.0
19
  tensorrt==10.8.0.43
20
  tensorrt_cu12==10.8.0.43
21
  tensorrt_cu12_bindings==10.8.0.43
22
  tensorrt_cu12_libs==10.8.0.43
23
- onnxruntime-gpu==1.18.0
 
13
  ultralytics-thop==2.0.14
14
  #opencv-python==4.11.0.86
15
  python-dotenv==1.0.0
 
16
  matplotlib==3.8.1
17
  onnx==1.17.0
18
  tensorrt==10.8.0.43
19
  tensorrt_cu12==10.8.0.43
20
  tensorrt_cu12_bindings==10.8.0.43
21
  tensorrt_cu12_libs==10.8.0.43
22
+ optimum[onnxruntime-gpu]==1.24.0