PyTorch下 CUDA 和 CuDNN 安装验证程序

装完gpu版的 pytorch，和给cuda装上cudnn后，不知道装对了没有，找了个验证程序验证一下

import torch
# 若正常则静默

a = torch.tensor(1.)
# 若正常则静默

a.cuda()
# 若正常则返回 tensor(1., device='cuda:0')

from torch.backends import cudnn
# 若正常则静默

cudnn.is_available()
# 若正常则返回 True

cudnn.is_acceptable(a.cuda()) 
# 若正常则返回 True

注意我的Pytorch版本是0.4.1，若是旧版本， 第二句话为：

a = torch.Tensor([1.]) #大写T，旧版本没有tensor这种形式，另外需加括号，旧版本没有0维tensor

a.cuda()
# 若正常则返回 tensor([ 1.], device='cuda:0')