mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-06 03:27:44 +00:00
Tim
a355b57623
* add runtime info * Update rdagent/scenarios/data_science/scen/__init__.py Co-authored-by: you-n-g <you-n-g@users.noreply.github.com> * reformatted by black --------- Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
41 lines
1022 B
Python
41 lines
1022 B
Python
import platform
|
|
import sys
|
|
from importlib.metadata import distributions
|
|
|
|
|
|
def print_runtime_info():
|
|
print(f"Python {sys.version} on {platform.system()} {platform.release()}")
|
|
|
|
|
|
def get_installed_packages():
|
|
return {dist.metadata["Name"].lower(): dist.version for dist in distributions()}
|
|
|
|
|
|
def print_filtered_packages(installed_packages, filtered_packages):
|
|
for package_name in filtered_packages:
|
|
version = installed_packages.get(package_name.lower())
|
|
if version:
|
|
print(f"{package_name}=={version}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print_runtime_info()
|
|
filtered_packages = [
|
|
"transformers",
|
|
"accelerate",
|
|
"torch",
|
|
"tensorflow",
|
|
"pandas",
|
|
"numpy",
|
|
"scikit-learn",
|
|
"scipy",
|
|
"lightgbm",
|
|
"vtk",
|
|
"opencv-python",
|
|
"keras",
|
|
"matplotlib",
|
|
"pydicom",
|
|
]
|
|
installed_packages = get_installed_packages()
|
|
print_filtered_packages(installed_packages, filtered_packages)
|