踩了巨多坑啊!!!!!真的不要想当然的以为把包下下来了就万事大吉了,一定要注意版本兼容的问题!!!
首先:查看NVIDIA驱动号还有cuda版本
nvidia-smi
显示如下界面,可以发现驱动器版本是430.26 Cuda版本是10.2
驱动器的安装和cuda的安装可以看看参照其它博客,这里不再叙述。
随后打开pytorch官网:https://pytorch.org/,
并往下滑至这个界面:
选择操作系统、package、python版本、cuda版本后,生成一条命令,随后复制命令行到终端运行即可。