use newer pynvml package

2023-08-09 21:06:28 +00:00
parent e303d64728
commit 9c314101d5
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,4 +19,4 @@ evaluate==0.4.0
 rouge-score==0.1.2
 scipy
 scikit-learn==1.2.2
-nvidia-ml-py3
+pynvml