对应版本安装cuda

如下

查看显卡驱动：


  nvidia-smi

PyTorch

PyTorch官方网站