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