Skip to content

GitLab

Unverified Commit c4be16e1 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

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

parent 3d8a5f06
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
collect_env.py
View file @ c4be16e1
...@@ -66,6 +66,8 @@ DEFAULT_CONDA_PATTERNS = { ...@@ -66,6 +66,8 @@ DEFAULT_CONDA_PATTERNS = {
"nccl", "nccl",
"transformers", "transformers",
"zmq", "zmq",
"nvidia",
"pynvml",
} }
DEFAULT_PIP_PATTERNS = { DEFAULT_PIP_PATTERNS = {
...@@ -79,6 +81,8 @@ DEFAULT_PIP_PATTERNS = { ...@@ -79,6 +81,8 @@ DEFAULT_PIP_PATTERNS = {
"nccl", "nccl",
"transformers", "transformers",
"zmq", "zmq",
"nvidia",
"pynvml",
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment