装完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')