1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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
|
|
