Spaces:
Sleeping
Sleeping
update requirements to use gpu
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -19,4 +19,5 @@ 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
|
|
|
|
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.19.2
|