pytorch 命令
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版本
好记性不如烂键盘---点滴、积累、进步!