一、检查版本是否匹配

使用Python进入控制台，然后执行：
 

>>>print(torch.cuda.is_available())

如果输出为 False，则说明不匹配

 

退出Python环境，再检查系统版本：

exit()

nvcc -V

二、安装

去官网 PyTorch 获取pip安装命令，按照可视化进行选择,复制下方的一键安装命令进行安装

选择适合自己的 平台 和版本，在控制台执行安装即可，注意在windows下使用的是pip不是pip3