about summary refs log tree commit diff
path: root/nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix')
-rw-r--r--nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix7
1 files changed, 6 insertions, 1 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix b/nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix
index a8c327a08076..b5c8f5a53d68 100644
--- a/nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix
+++ b/nixpkgs/pkgs/development/python-modules/nvidia-ml-py/default.nix
@@ -1,6 +1,8 @@
 { lib
 , fetchPypi
 , buildPythonPackage
+, substituteAll
+, addOpenGLRunpath
 }:
 
 buildPythonPackage rec {
@@ -15,7 +17,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;
+    })
   ];
 
   # no tests