From c63c33bc67dc8a590f1541911c8c0ca2d8b68696 Mon Sep 17 00:00:00 2001 From: Dmitry Duev Date: Thu, 15 Aug 2024 11:19:08 -0700 Subject: [PATCH] Set LIB_PATH to libnvidia-ml.so.1 instead of libnvidia-ml.so on Linux In the official Go bindings for NVML, they use libnvidia-ml.so.1: https://github.com/NVIDIA/go-nvml/blob/0e815c71ca6e8184387d8b502b2ef2d2722165b9/pkg/nvml/lib.go#L30 --- nvml-wrapper/src/lib.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nvml-wrapper/src/lib.rs b/nvml-wrapper/src/lib.rs index 967e66a..c9fc7be 100644 --- a/nvml-wrapper/src/lib.rs +++ b/nvml-wrapper/src/lib.rs @@ -155,7 +155,7 @@ use crate::bitmasks::InitFlags; const LIB_PATH: &str = "nvml.dll"; #[cfg(target_os = "linux")] -const LIB_PATH: &str = "libnvidia-ml.so"; +const LIB_PATH: &str = "libnvidia-ml.so.1"; /// Determines the major version of the CUDA driver given the full version. ///