目录 1. 检查cuda2. 检查cudnn 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