目录
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