Package ai.djl.pytorch.jni
Class LibUtils
java.lang.Object
ai.djl.pytorch.jni.LibUtils
Utilities for finding the PyTorch Engine binary on the System.
The Engine will be searched for in a variety of locations in the following order:
- In the path specified by the PYTORCH_LIBRARY_PATH environment variable
- In a jar file location in the classpath. These jars can be created with the pytorch-native module.
-
Method Details
-
loadLibrary
public static void loadLibrary() -
getVersion
-
getLibtorchPath
-