踩了不少坑,给大家po一个合理的安装顺序:
安装Anaconda
最好先在anaconda prompt里new一个conda的python3.6虚拟环境(方法去别的文章搜一下),再整理一下conda的pkg文件夹确保你以前没有安装过跟torch以及cuda toolkit相关的包。如果有的话,建议重装一下anaconda,或者clean一下list里的包。因为这个我配环境配了好久,出了很多问题。
CUDA Toolkit
先更新你的显卡驱动,现在nvidia的最新驱动好像是集合了,适配多张显卡的,更新完毕后。
打开prompt,查看CUDA适配的最高版本
nvidia-smi
如图,低于496.13以及11.5的cuda版本都是可以安装的。
然后先去pytorch官网上找你要安装的torch版本
如图,版本是稳定版的torch1.10。CUDA是10.2,最好严格按照10.2安装CUDA,11.3也一样。先复制下面的命令行备用。