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