kiurtis commited on
Commit
5d42902
·
1 Parent(s): d80f0d9

update requirements to use gpu

Browse files
Files changed (1) hide show
  1. 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