pytorchがcudaをサポートしているかどうかを確認する

1. cudaを確認する

import torch
torch.cuda.is_available()   # 检查cuda是否可用
torch.version.cuda          # 查看cuda版本

2. cudnn を確認する

torch.backends.cudnn.is_available()  # 检查cudnn是否可用
torch.backends.cudnn.version()       # 查看cudnn版本

参考リンク：
https ://blog.csdn.net/yang332233/article/details/125752424