[pytorch] Fix load native library failure on Android (#1485)

Change-Id: Ic255c90bcba5a99aa467a104305f5fe4d7af5620
deepjavalibrary · Jan 28, 2022 · 4655660 · 4655660
1 parent 98d35e1
commit 4655660
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java
@@ -65,7 +65,7 @@ public static synchronized void loadLibrary() {
         // TODO workaround to make it work on Android Studio
         // It should search for several places to find the native library
         if ("http://www.android.com/".equals(System.getProperty("java.vendor.url"))) {
-            System.loadLibrary(JNI_LIB_NAME); // NOPMD
+            System.loadLibrary("djl_torch"); // NOPMD
             return;
         }
         libTorch = getLibTorch();