pkgs/development/python-modules/nvidia-ml-py/0001-locate-libnvidia-ml.so.1-on-NixOS.patch at master · pyrox.dev/nixpkgs

pyrox.dev / nixpkgs

lol

nixpkgs / pkgs / development / python-modules / nvidia-ml-py / 0001-locate-libnvidia-ml.so.1-on-NixOS.patch

at master 823 B view raw

 1diff --git a/pynvml.py b/pynvml.py
 2index 9a424de..669afe0 100644
 3--- a/pynvml.py
 4+++ b/pynvml.py
 5@@ -1676,7 +1676,11 @@ def _LoadNvmlLibrary():
 6                             nvmlLib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
 7                     else:
 8                         # assume linux
 9-                        nvmlLib = CDLL("libnvidia-ml.so.1")
10+                        try:
11+                            nvmlLib = CDLL("libnvidia-ml.so.1")
12+                        except OSError:
13+                            # Assume NixOS
14+                            nvmlLib = CDLL("@driverLink@/lib/libnvidia-ml.so.1")
15                 except OSError as ose:
16                     _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
17                 if (nvmlLib == None):