ai.djl.pytorch.jni

Class LibUtils

java.lang.Object

ai.djl.pytorch.jni.LibUtils

public final class LibUtils
extends java.lang.Object

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:

1. In the path specified by the PYTORCH_LIBRARY_PATH environment variable
2. In a jar file location in the classpath. These jars can be created with the pytorch-native module.

Method Summary

Modifier and Type	Method and Description
static java.lang.String	getLibName(java.lang.String djlVersion, java.lang.String jniVersion)
static java.lang.String	getLibtorchPath()
static java.lang.String	getVersion()
static void	loadLibrary()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

loadLibrary

public static void loadLibrary()

getLibName

public static java.lang.String getLibName(java.lang.String djlVersion,
                                          java.lang.String jniVersion)

getVersion

public static java.lang.String getVersion()

getLibtorchPath

public static java.lang.String getLibtorchPath()