force onnxruntime_gpu to 1.16.3

Signed-off-by: Yaliang Wu <[email protected]>
rbhavna · Jun 20, 2024 · 5e347f4 · 5e347f4
1 parent 22d6ed9
commit 5e347f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle
@@ -53,11 +53,11 @@ dependencies {
     //arm/macos doesn't support GPU
     if (os.macOsX || System.getProperty("os.arch") == "aarch64") {
         dependencies {
-            implementation "com.microsoft.onnxruntime:onnxruntime:1.17.1"
+            implementation "com.microsoft.onnxruntime:onnxruntime:1.16.3!!"
         }
     } else {
         dependencies {
-            implementation "com.microsoft.onnxruntime:onnxruntime_gpu:1.17.1"
+            implementation "com.microsoft.onnxruntime:onnxruntime_gpu:1.16.3!!"
         }
     }