diff --git a/requirements.txt b/requirements.txt
index 0d36af5..f51fc17 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,5 @@ tensorflow==2.13.0rc1; sys_platform == 'darwin'
 tensorflow==2.12.0; sys_platform != 'darwin'
 opennsfw2==0.10.2
 protobuf==4.23.2
-tqdm==4.65.0
\ No newline at end of file
+pynvml==11.5.0
+tqdm==4.65.0