Upgrade onnxruntime to 1.18.1

vietanhdev · Aug 3, 2024 · b5f8901 · b5f8901
1 parent 62f0a24
commit b5f8901
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/anylabeling/app_info.py b/anylabeling/app_info.py
@@ -1,4 +1,4 @@
 __appname__ = "AnyLabeling"
 __appdescription__ = "Effortless data labeling with AI support"
-__version__ = "0.4.1"
+__version__ = "0.4.2"
 __preferred_device__ = "CPU"  # GPU or CPU
diff --git a/requirements-gpu.txt b/requirements-gpu.txt
@@ -5,6 +5,6 @@ natsort==8.1.0
 termcolor==1.1.0
 PyYAML==6.0
 onnx==1.16.2
-onnxruntime-gpu==1.16.3
+onnxruntime-gpu==1.18.1
 qimage2ndarray==1.10.0
 darkdetect==0.8.0
diff --git a/requirements-macos.txt b/requirements-macos.txt
@@ -5,6 +5,6 @@ natsort==8.1.0
 termcolor==1.1.0
 PyYAML==6.0
 onnx==1.16.2
-onnxruntime==1.16.3
+onnxruntime==1.18.1
 qimage2ndarray==1.10.0
 darkdetect==0.8.0
diff --git a/requirements.txt b/requirements.txt
@@ -5,6 +5,6 @@ natsort==8.1.0
 termcolor==1.1.0
 PyYAML==6.0
 onnx==1.16.2
-onnxruntime==1.16.3
+onnxruntime==1.18.1
 qimage2ndarray==1.10.0
 darkdetect==0.8.0
diff --git a/setup.py b/setup.py
@@ -60,10 +60,10 @@ def get_install_requires():
     # Note: onnxruntime-gpu is not available on macOS
     preferred_device = get_preferred_device()
     if preferred_device == "GPU" and platform.system() != "Darwin":
-        install_requires.append("onnxruntime-gpu==1.16.3")
+        install_requires.append("onnxruntime-gpu==1.18.1")
         print("Building AnyLabeling with GPU support")
     else:
-        install_requires.append("onnxruntime==1.16.3")
+        install_requires.append("onnxruntime==1.18.1")
         print("Building AnyLabeling without GPU support")
 
     return install_requires