Fix onnx dependencies in requirements.txt (#43)

nvidia-riva · May 6, 2024 · cc4f3a8 · cc4f3a8
1 parent d476530
commit cc4f3a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -4,5 +4,7 @@
 nemo_toolkit>=1.6.0
 nvidia-eff>=0.6.4
 nvidia-eff-tao-encryption>=0.1.8
-onnxruntime-gpu>=1.13.1
+onnx==1.14.1
+onnxruntime==1.16.3
+onnxruntime-gpu==1.16.3
 onnx_graphsurgeon