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