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 index 3395cbb7140..1a7c5d4aca8 100644 --- 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();