gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

template patch with addOpenGLRunpath

Browse Source
This commit is contained in:
David Wilemski
2023-01-07 16:44:45 +00:00
committed by Florian Brandes
parent 2ed170dbf4
commit aefa87d9dc
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/pynvml/0001-locate-libnvidia-ml.so.1-on-NixOS.patch
View File

@@ -11,7 +11,7 @@ index 56d908f..1de0b97 100644
+ nvmlLib = CDLL("libnvidia-ml.so.1") + nvmlLib = CDLL("libnvidia-ml.so.1")
+ except OSError: + except OSError:
+ # assume NixOS + # assume NixOS
+ nvmlLib = CDLL("/run/opengl-driver/lib/libnvidia-ml.so.1") + nvmlLib = CDLL("@driverLink@/lib/libnvidia-ml.so.1")
except OSError as ose: except OSError as ose:
_nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND) _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
if (nvmlLib == None): if (nvmlLib == None):

7
pkgs/development/python-modules/pynvml/default.nix
View File

@@ -1,8 +1,10 @@
{ lib { lib
, buildPythonPackage , buildPythonPackage
, fetchPypi , fetchPypi
, substituteAll
, pythonOlder , pythonOlder
, cudatoolkit , cudatoolkit
, addOpenGLRunpath
}: }:
buildPythonPackage rec { buildPythonPackage rec {
@@ -16,7 +18,10 @@ buildPythonPackage rec {
}; };
patches = [ patches = [
./0001-locate-libnvidia-ml.so.1-on-NixOS.patch (substituteAll {
src = ./0001-locate-libnvidia-ml.so.1-on-NixOS.patch;
inherit (addOpenGLRunpath) driverLink;
})
]; ];
propagatedBuildInputs = [ cudatoolkit ]; propagatedBuildInputs = [ cudatoolkit ];