踩了不少坑,给大家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也一样。先复制下面的命令行备用。
这儿贴一个CUDAtoolkit的旧版本列表,如果是10.2,找到10.2下载并安装。(安装时最好只自定义安装CUDA,把显卡驱动部分取消勾选)Archive of Previous CUDA Releases
安装完毕后下载CUDNN这里选最新的适配CUDA10.2的CUDNN就可以,下载后解压复制进CUDA的文件夹里覆盖就算安装成功。
conda prompt里输入查看cuda安装情况
nvcc -V
如图可以查看安装情况
conda激活一下你new的虚拟环境,然后复制上面备用的torch安装命令等待包下载安装即可,速度慢的话网上搜一个清华源或者挂代理,另外确保你的网络连接比较稳定,不然安装容易出问题。
包下载完以后好像会有三个确认,我之前因为conda的包列表路径不干净,cudatoolkit安装老是出问题,搜了不少文章有用pip wheel强制安装的,但是我搜不到对应的轮子文件,也有说是版本不对应的,试了很多次,只有重装conda这个方法合适。
安装完毕后进入python
python import torch torch.cuda.is_available()
显示true则torch安装成功
有别的问题可以评论区问我。



