collect_env.py: get conda magma and mkl information (#14854)

Summary:
Fixes #12371
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14854

Differential Revision: D13363635

Pulled By: zou3519

fbshipit-source-id: f8b5d05038bf5ce451399dfeed558ae298178128

diff --git a/torch/utils/collect_env.py b/torch/utils/collect_env.py
index 32b47ee..214c64c 100644 (file)
--- a/torch/utils/collect_env.py
+++ b/torch/utils/collect_env.py
@@ -66,9 +66,9 @@ def run_and_parse_first_match(run_lambda, command, regex):
 
 def get_conda_packages(run_lambda):
     if get_platform() == 'win32':
-        grep_cmd = r'findstr /R "torch soumith"'
+        grep_cmd = r'findstr /R "torch soumith mkl magma"'
     else:
-        grep_cmd = r'grep "torch\|soumith"'
+        grep_cmd = r'grep "torch\|soumith\|mkl\|magma"'
     out = run_and_read_all(run_lambda, 'conda list | ' + grep_cmd)
     if out is None:
         return out