onnxruntime depends on architecture, not os

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Mar 12, 2024 · 1a0af3b · 1a0af3b
1 parent 2e0946a
commit 1a0af3b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle
@@ -51,9 +51,8 @@ dependencies {
     implementation("ai.djl.onnxruntime:onnxruntime-engine:0.21.0") {
         exclude group: "com.microsoft.onnxruntime", module: "onnxruntime"
     }
-    def os = DefaultNativePlatform.currentOperatingSystem
-    //mac doesn't support GPU
-    if (os.macOsX) {
+    //arm doesn't support GPU
+    if (System.getProperty("os.arch") == "aarch64") {
         dependencies {
             implementation "com.microsoft.onnxruntime:onnxruntime:1.14.0"
         }
@@ -122,4 +121,4 @@ spotless {
 
         eclipse().configFile rootProject.file('.eclipseformat.xml')
     }
-}
+}