cleanup GPU stats dependencies

FedML-AI · Sep 9, 2023 · 7dd59dc · 7dd59dc
1 parent a351737
commit 7dd59dc
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -27,7 +27,6 @@ def finalize_options(self):
     "wget",
     "paho-mqtt",
     "boto3",
-    "pynvml",
     "scikit-learn",
     "networkx<3.0",
     "click",
@@ -37,7 +36,6 @@ def finalize_options(self):
     "gensim",
     "multiprocess",
     "smart-open==6.3.0",
-    "nvidia-ml-py3",
     "matplotlib",
     "dill",
     "pandas",

diff --git a/requirements.txt b/requirements.txt
@@ -5,7 +5,6 @@ tqdm~=4.64.0
 wget~=3.2
 paho-mqtt~=1.6.1
 boto3~=1.24.32
-pynvml
 scikit-learn~=1.1.1
 networkx~=2.8.5
 click~=8.1.3
@@ -15,7 +14,6 @@ spacy~=3.4.0
 gensim~=4.2.0
 multiprocess~=0.70.13
 smart-open==6.3.0
-nvidia-ml-py3
 matplotlib~=3.5.2
 dill~=0.3.5.1
 pandas~=1.4.3