Add NVML python bindings

cms-sw · Apr 17, 2024 · 8cffe60 · 8cffe60
1 parent dcee9d7
commit 8cffe60
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pip/requirements.txt b/pip/requirements.txt
@@ -214,6 +214,7 @@ neurolab==0.3.5
 node-semver==0.9.0
 notebook==6.5.3
 notebook-shim==0.2.2
+nvidia-ml-py==12.535.133
 # numba==0.56.4
 numexpr==2.8.4
 # setuptools version <64 is needed by numpy: https://github.com/pypa/setuptools/issues/3549

diff --git a/python_tools.spec b/python_tools.spec
@@ -54,6 +54,7 @@ Requires: py3-mistune
 Requires: py3-nbconvert
 Requires: py3-nbformat
 Requires: py3-notebook
+Requires: py3-nvidia-ml-py
 Requires: py3-packaging
 Requires: py3-pandocfilters
 Requires: py3-pathlib2