1.选择相应的系统与cuda版本
直接使用pip安装，此处挂梯子可能会快一点。
理论上直接安装就完成了，CUDA和cuDNN应该会一步到位自动安装完成。
2.检验是否安装成功
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
正常则如图所示