Spaces:
Sleeping
Sleeping
maybe fix the gpu onnx thing
Browse files- 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.
|
|
|
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
|