diff --git a/requirements.base.txt b/requirements.base.txt
index 67ba004..e7a302f 100644
--- a/requirements.base.txt
+++ b/requirements.base.txt
@@ -8,4 +8,5 @@ loguru
 pyzmq
 gitpython
 gdown
-lap
\ No newline at end of file
+lap
+pynvml
\ No newline at end of file