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

c4be16e1 · youkaichao · GitHub · 3d8a5f06 · c4be16e1
Unverified Commit c4be16e1 authored Aug 19, 2024 by youkaichao Committed by GitHub Aug 19, 2024
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

collect_env.py collect_env.py +4 -0

No files found.
--- a/collect_env.py
+++ b/collect_env.py
@@ -66,6 +66,8 @@ DEFAULT_CONDA_PATTERNS = {
    "nccl",
    "transformers",
    "zmq",
+    "nvidia",
+    "pynvml",
 }

 DEFAULT_PIP_PATTERNS = {
@@ -79,6 +81,8 @@ DEFAULT_PIP_PATTERNS = {
    "nccl",
    "transformers",
    "zmq",
+    "nvidia",
+    "pynvml",
 }