From 8cffe6065bf0aed75ea09a5e6899076436446c6e Mon Sep 17 00:00:00 2001
From: Andrea Bocci <andrea.bocci@cern.ch>
Date: Sat, 13 Apr 2024 09:11:11 +0200
Subject: [PATCH] Add NVML python bindings

---
 pip/requirements.txt | 1 +
 python_tools.spec    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/pip/requirements.txt b/pip/requirements.txt
index 55cf94df16f..fc3c5b6a5a9 100644
--- 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
index f01fd542364..433db793d25 100644
--- 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