Попытка добавить onnx в работе с nvidia

2026-04-15 17:53:06 +05:00
parent 0c871c2314
commit 7addcf051c
13 changed files with 2484 additions and 371 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,14 @@ requires-python = ">=3.12"
 dependencies = [
    "imageio>=2.37.3",
    "numpy>=2.4.4",
+    "nvidia-modelopt[all]>=0.33.1",
    "omegaconf>=2.3.0",
+    "onnx>=1.21.0",
+    "onnxscript>=0.6.2",
    "opencv-python>=4.13.0.92",
-    "torch>=2.11.0",
+    "tensorrt>=10.16.1.11",
+    "torch==2.5.1",
+    "torch-tensorrt>=2.5.0",
+    "torchvision>=0.20.1",
    "tqdm>=4.67.3",
 ]