From 0b9e1e1ecfdc86cea6a7222ae2c2140e8faa994d Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Mon, 18 Dec 2023 09:40:43 -0800 Subject: [PATCH] [api] Handles both JNA conflict and missing case --- api/src/main/java/ai/djl/util/cuda/CudaUtils.java | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/api/src/main/java/ai/djl/util/cuda/CudaUtils.java b/api/src/main/java/ai/djl/util/cuda/CudaUtils.java index edcbf40eef3..b30a208f6ab 100644 --- a/api/src/main/java/ai/djl/util/cuda/CudaUtils.java +++ b/api/src/main/java/ai/djl/util/cuda/CudaUtils.java @@ -262,16 +262,14 @@ private static CudaLibrary loadLibrary() { } catch (UnsatisfiedLinkError e) { logger.debug("cudart library not found."); logger.trace("", e); - return null; - } catch (IncompatibleClassChangeError e) { + } catch (LinkageError e) { logger.warn("You have a conflict version of JNA in the classpath."); logger.debug("", e); - return null; } catch (SecurityException e) { logger.warn("Access denied during loading cudart library."); logger.trace("", e); - return null; } + return null; } private static String[] execute(int deviceId) {