Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
830 B · 38 lines
Nix
at litex
123456789101112131415161718192021222324252627282930313233343536373839{ lib, buildPythonPackage, fetchPypi, substituteAll, pythonOlder, cudatoolkit, addOpenGLRunpath}:
buildPythonPackage rec { pname = "pynvml"; version = "11.5.0"; disabled = pythonOlder "3.6";
src = fetchPypi { inherit pname version; hash = "sha256-0CeyG5WxCIufwngRf59ht8Z/jjOnh+n4P3NfD3GsMtA="; };
patches = [ (substituteAll { src = ./0001-locate-libnvidia-ml.so.1-on-NixOS.patch; inherit (addOpenGLRunpath) driverLink; }) ];
propagatedBuildInputs = [ cudatoolkit ];
doCheck = false; # no tests in PyPi dist pythonImportsCheck = [ "pynvml" "pynvml.smi" ];
meta = with lib; { description = "Python bindings for the NVIDIA Management Library"; homepage = "https://www.nvidia.com"; license = licenses.bsd3; maintainers = [ maintainers.bcdarwin ]; };}