安装anaconda以及anaconda安装pytorch踩坑分享
本文主要是说明一些坑,对安装步骤,主要转载一些前辈们的经验,建议读者可以先通篇看完,对可能出现的问题心里面有个大概的思路,然后再进行安装
anaconda和pytorch安装教程
conda下载PyTorch速度慢:国内源,添加国内源和删源以及直接下载等操作
anaconda查看、添加、删除镜像源
1、注意查看自己电脑的CUDA
这个主要是关系到安装pytorch时Compute Platform
的选择问题,如果安装的版本不匹配,即使import torch
没有报错,也会在torch.cuda.is_available()
时返回false
可以在cmd中输入命令NVIDIA-smi
查看
举例:我的是10.1版本,但是pytorch官网直接给出的是10.2和11.3的,所以我就要查看历史版本
我是全局搜索了CUDA 10.1
然后找到了自己需要的版本,在自己创建的环境中输入这个命令(安装命令这里也有个注意事项,在后面说明)
2、安装pytorch时即使配置了镜像源也很慢,大概率还会安装失败
解决:去掉上个注意里面官网给出的安装pytorch的命令末尾的 -c pytorch
这样你的安装就会自动去配置的镜像里面找你需要的包了
3、注意conda安装信息里面的警告,我安装时候是警告我要更新conda,不更新下载的也是很慢,大概率会出错
最好是根据警告里面的命令进行更新,conda update -n base -c defaults conda