之前装tensorflow时候,看网上的教程基本上都是先去nvidia官网下载对应NVIDIA驱动版本的cuda以及它的更新包和对应的cudnn,然后一个包一个包的安装,按照教程走也基本没问题,最大的困难就是网速了,不过根据个人经验,一般早上下载服务器在国外的资源,速度会相对较快,但还是麻烦,对新手来说,不够友好。
后来发现可以使用conda安装tensorflow,它会自动帮你解决依赖项问题,不用自己再另外安装cuda和cudnn以及其他依赖包,非常简单粗暴,我很喜欢,哈哈。
最近又想装下pytorch,也想使用conda一键安装,但是遇到了一个意外,死活下不下来pytorch官网的包,最终发现pytorch官网上的那句conda安装pytorch命令,已经指定了pytorch包的下载源是官网的服务器,难怪一直下不下来,然后我就自己添加了清华镜像源,发现两分钟就下好了。所以把这个记下来,顺便分享给大家。
下面是具体步骤:
1,添加conda的清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set