18 lines
843 B
Diff
18 lines
843 B
Diff
|
diff --git a/pynvml/nvml.py b/pynvml/nvml.py
|
||
|
|
index 56d908f..1de0b97 100644
|
||
|
|
--- a/pynvml/nvml.py
|
||
|
|
+++ b/pynvml/nvml.py
|
||
|
|
@@ -1475,7 +1475,11 @@ def _LoadNvmlLibrary():
|
||
|
|
nvmlLib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
|
||
|
|
else:
|
||
|
|
# assume linux
|
||
|
|
- nvmlLib = CDLL("libnvidia-ml.so.1")
|
||
|
|
+ try:
|
||
|
|
+ nvmlLib = CDLL("libnvidia-ml.so.1")
|
||
|
|
+ except OSError:
|
||
|
|
+ # assume NixOS
|
||
|
|
+ nvmlLib = CDLL("@driverLink@/lib/libnvidia-ml.so.1")
|
||
|
|
except OSError as ose:
|
||
|
|
_nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
|
||
|
|
if (nvmlLib == None):
|