Anton Bushuiev commited on
Commit
3dd7c16
·
1 Parent(s): 64f405b

Print environment packages on start

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -44,6 +44,21 @@ from ppiformer.utils.torch import fill_diagonal
44
  from ppiformer.definitions import PPIFORMER_WEIGHTS_DIR
45
 
46
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
47
  logging.basicConfig(
48
  level=logging.INFO,
49
  format='%(asctime)s - %(levelname)s - %(message)s',
 
44
  from ppiformer.definitions import PPIFORMER_WEIGHTS_DIR
45
 
46
 
47
+ import pkg_resources
48
+ import sys
49
+
50
+ def print_package_versions():
51
+ installed_packages = sorted([f"{pkg.key}=={pkg.version}" for pkg in pkg_resources.working_set])
52
+ print("Installed packages and their versions:")
53
+ for package in installed_packages:
54
+ print(package)
55
+
56
+ print("\nPython version:")
57
+ print(sys.version)
58
+
59
+ print_package_versions()
60
+
61
+
62
  logging.basicConfig(
63
  level=logging.INFO,
64
  format='%(asctime)s - %(levelname)s - %(message)s',