mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
17 lines
468 B
Python
17 lines
468 B
Python
import torch
|
|
|
|
|
|
def pytorch_cudnn_version() -> str:
|
|
message = (
|
|
f"pytorch.version={torch.__version__}, "
|
|
f"cuda.available={torch.cuda.is_available()}, "
|
|
)
|
|
|
|
if torch.backends.cudnn.enabled:
|
|
message += (
|
|
f"cudnn.version={torch.backends.cudnn.version()}, "
|
|
f"cudnn.benchmark={torch.backends.cudnn.benchmark}, "
|
|
f"cudnn.deterministic={torch.backends.cudnn.deterministic}"
|
|
)
|
|
return message
|