使用 PyTorch 查看 CUDA 和 cuDNN 版本
参考连接Linux查看CUDA和cuDNN版本 - 胖墩哥 - 博客园 (cnblogs.com)
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
参考连接Linux查看CUDA和cuDNN版本 - 胖墩哥 - 博客园 (cnblogs.com)
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())