646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
18 lines
823 B
Diff
18 lines
823 B
Diff
diff --git a/pynvml.py b/pynvml.py
|
|
index 9a424de..669afe0 100644
|
|
--- a/pynvml.py
|
|
+++ b/pynvml.py
|
|
@@ -1676,7 +1676,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):
|