解决方法
1.测试
import torch
from torch.backends import cudnn
# 若正常则静默
a = torch.tensor(1.)
# 若正常则静默
a.cuda()
# 若正常则返回
tensor(1., device='cuda:0')
# 若正常则静默
cudnn.is_available()
# 若正常则返回 True
cudnn.is_acceptable(a.cuda())
# 若正常则返回 true
2.添加代码 在你的模型运行之前
A不在使用cudnn
torch.backends.cudnn.enabled = False
B坚持使用
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
最终ok