Update env.py (#1635)

This commit is contained in:
zr_jin 2024-05-22 22:29:38 +08:00 committed by GitHub
parent 1adf1e441d
commit 42a97f6d7b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -108,7 +108,7 @@ def get_env_info() -> Dict[str, Any]:
"torch-version": str(torch.__version__),
"torch-cuda-available": torch.cuda.is_available(),
"torch-cuda-version": torch.version.cuda,
"python-version": sys.version[:3],
"python-version": ".".join(sys.version.split(".")[:2]),
"icefall-git-branch": get_git_branch_name(),
"icefall-git-sha1": get_git_sha1(),
"icefall-git-date": get_git_date(),