Commit 7650f5f8 authored by David Wilemski's avatar David Wilemski
Browse files

template patch with addOpenGLRunpath

parent 27513ed3

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

+1 −1
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@ index 56d908f..1de0b97 100644 
+                            nvmlLib = CDLL("libnvidia-ml.so.1")

+                        except OSError:

+                            # assume NixOS

+                            nvmlLib = CDLL("/run/opengl-driver/lib/libnvidia-ml.so.1")

+                            nvmlLib = CDLL("@driverLink@/lib/libnvidia-ml.so.1")

                 except OSError as ose:

                     _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)

                 if (nvmlLib == None):

pkgs/development/python-modules/pynvml/default.nix

+6 −1
Original line number Diff line number Diff line 
{ lib

, buildPythonPackage

, fetchPypi

, substituteAll

, pythonOlder

, cudatoolkit

, addOpenGLRunpath

}:



buildPythonPackage rec {
@@ -16,7 +18,10 @@ buildPythonPackage rec { 
  };



  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 ];