print additional info when starting ipi client

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(