pkgs/development/python-modules/nvidia-ml-py/default.nix

   1 { lib
   2 , fetchPypi
   3 , buildPythonPackage
   4 , substituteAll
   5 , addOpenGLRunpath
   6 }:
   7
   8 buildPythonPackage rec {
   9   pname = "nvidia-ml-py";
  10   version = "12.535.133";
  11   format = "setuptools";
  12
  13   src = fetchPypi {
  14     inherit pname version;
  15     extension = "tar.gz";
  16     hash = "sha256-sVWa8NV90glVv1jQWv/3sWbd1ElH6zBRyZBWOHmesdw=";
  17   };
  18
  19   patches = [
  20     (substituteAll {
  21       src = ./0001-locate-libnvidia-ml.so.1-on-NixOS.patch;
  22       inherit (addOpenGLRunpath) driverLink;
  23     })
  24   ];
  25
  26   # no tests
  27   doCheck = false;
  28
  29   pythonImportsCheck = [ "pynvml" ];
  30
  31   meta = {
  32     description = "Python Bindings for the NVIDIA Management Library";
  33     homepage = "https://pypi.org/project/nvidia-ml-py";
  34     license = lib.licenses.bsd3;
  35     platforms = [ "x86_64-linux" ];
  36     maintainers = with lib.maintainers; [ GaetanLepage ];
  37   };
  38 }