Use torch.utils.collect_env in run_experiments.py

Author: voltjia

run_experiments.py

@@ -100,6 +100,11 @@ def _benchmark_ops(ops, *args, **kwargs):
     random.seed(0)
     torch.manual_seed(0)
 
+    with open("torch.utils.collect_env.log", "w") as f:
+        subprocess.run(
+            ("python", "-m", "torch.utils.collect_env"), stdout=f, check=True
+        )
+
     radon_commands = (
         (
             "radon",
@@ -184,6 +189,4 @@ def _benchmark_ops(ops, *args, **kwargs):
                 )
 
     with open("performance_evaluation.tex", "w") as f:
-        subprocess.run(
-            ("python", "evaluate_performance.py"), stdout=f, check=True
-        )
+        subprocess.run(("python", "evaluate_performance.py"), stdout=f, check=True)