add files

funasr_local/torch_utils/pytorch_version.py

@@ -0,0 +1,16 @@
+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