change onnxruntime to onnxruntime-gpu

supervisely-ecosystem · Nov 26, 2024 · 054e55d · 054e55d
1 parent cb52462
commit 054e55d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/serve/requirements.txt b/serve/requirements.txt
@@ -4,5 +4,5 @@
 # --extra-index-url https://download.pytorch.org/whl/cu113
 # torch==1.10.1+cu113
 # torchvision==0.11.2+cu113
-git+https://github.com/supervisely/supervisely.git@sem_seg_benchmark
-onnxruntime==1.18.0
+onnxruntime-gpu==1.18.0
+git+https://github.com/supervisely/supervisely.git@sem_seg_benchmark