From 0a1511ef8df2ce3ab7dcb7d4571a86574cb924c7 Mon Sep 17 00:00:00 2001
From: Sander Vandenhaute <svdenhau@gmail.com>
Date: Fri, 5 Jul 2024 11:02:50 -0400
Subject: [PATCH] print additional info when starting ipi client

---
 psiflow/tools/client.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/psiflow/tools/client.py b/psiflow/tools/client.py
index 0938cd7..7566b1a 100644
--- a/psiflow/tools/client.py
+++ b/psiflow/tools/client.py
@@ -1,3 +1,4 @@
+import os
 import argparse
 from pathlib import Path
 
@@ -9,6 +10,9 @@
 
 
 if __name__ == "__main__":
+    print('OS environment values:')
+    for key, value in os.environ.items():
+        print(key, value)
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "--path_hamiltonian",
@@ -56,6 +60,10 @@
         dtype=args.dtype,
         max_force=args.max_force,
     )
+    import torch
+    import psutil
+    print('torch num threads: ', torch.get_num_threads())
+    print('cpu count: ', psutil.cpu_count(logical=False))
 
     try:
         run_driver(