Update env.py

This commit is contained in:
jinzr 2024-05-22 17:07:09 +08:00
parent 68980c5d0a
commit 62eb61037d

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(),