From c4be16e1a70d50b781038990087a717fd8834d4a Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@gmail.com>
Date: Mon, 19 Aug 2024 23:22:49 -0700
Subject: [PATCH] [misc] add nvidia related library in collect env (#7674)

---
 collect_env.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/collect_env.py b/collect_env.py
index 76df97b099b1b..839d54172e775 100644
--- 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",
 }