[misc] add nvidia related library in collect env (#7674)

vllm-project · Aug 20, 2024 · c4be16e · c4be16e
1 parent 3d8a5f0
commit c4be16e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/collect_env.py b/collect_env.py
@@ -66,6 +66,8 @@
     "nccl",
     "transformers",
     "zmq",
+    "nvidia",
+    "pynvml",
 }
 
 DEFAULT_PIP_PATTERNS = {
@@ -79,6 +81,8 @@
     "nccl",
     "transformers",
     "zmq",
+    "nvidia",
+    "pynvml",
 }