查看Pytorch是否安裝成功解決辦法

import torch    # 如正常則靜默
a = torch.Tensor([1.])    # 如正常則靜默
a.cuda()    # 如正常則返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常則靜默
cudnn.is_acceptable(a.cuda())    # 如正常則返回 "True"