From 2606be04ba33f228d4d7fa605c60430b2964557d Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Sun, 4 Jun 2023 13:24:32 +0530 Subject: [PATCH] add pynvml --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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