Anaconda安装 torch

最近在做一些深度学习的项目,需要配置gpu环境,这里把遇到的一些坑和解决方法记录下来,供大家借鉴,同时也是以后自己需要的时候可以回头看看。

关于pip安装和conda安装。简单来说pip也只是一个库,相当于一个安装工具。pip安装需要在cmd命令行中进行,方法为pip install +名字。但是利用pip安装网速极慢,是用的国外的channel,大概只有几kb/s,经常容易出现read time out的error。因此强烈建议使用conda install +名字。首先要把channel改为清华的镜像源,这样可以达到几mb/s的速度(之前因为版权问题关掉了一阵了,现在版权谈妥又开放了,这种有时效性的事情,网上就有坑让你删除这个镜像源,但现在不要删)。方法是在anaconda promote中base(root)打开terminal(还没安装anaconda的看下面图2那里先安装),进入命令行,输入以下:

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/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

如果需要安装最新的torch,再加入这个channel:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

windows系统最后需要加上这个来生成.condarc的文件:

conda config --set show_channel_urls yes

需要注意的是一定要把Navigator界面中channels的default删掉,如图1。删完记得点update channels!!!

Anaconda安装 torch
图1 删掉channels中default

安装anaconda的话直接下载安装包就行,如图2。下载地址很多,实在找不到私信我发你。

Anaconda安装 torch
图2 anaconda安装包

安装完记得添加环境变量,就算安装的时候点了自动添加也再看一下。如果没有,就手动添加:环境变量->系统变量->Path,如图3。具体取决于你安装在哪个盘C,D,E...没框的那几个好像是当时试坑加的。

Anaconda安装 torch
图3 环境变量

接着创建自己的环境,不要用base(root)这个,比如我创建的py3.7-gpu。创建方法为:打开base(root)的terminal进入命令行,运行conda clean -i清除索引缓存,保证用的是清华镜像站提供的索引。这里备注一下,你在anaconda navigator中看到的packages都是根据你设置的channel提供的,比如说如果没有添加

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

那么搜索torchvision可能就没有结果,如图4。之前说pip坑也是因为我用pip安装的,pip list和conda list都有,但是anaconda navigator中没有,所以等于零。

Anaconda安装 torch
图4 搜索torchvision

运行完

之后再运行

conda create -n py3.7-gpu python=3.7

就行,当然你也可以用3.6的版本如

conda create -n py3.6-gpu python=3.6

-n是name的缩写。我写这篇文章的时候刚出3.8但还是3.7 3.6靠谱一点,版本太新可能没有配套的tensorflow或torch。

如果顺利,就成功创建了一个environment,名字叫py3.7-gpu。创建的过程中可以在命令行看到从哪下载的,如果是正常清华的镜像源几秒钟就创建好了。接下来就是给这个environment装各种package了,方法都是在这个environment的terminal下输入conda install +名字。比如说spyder,tensorflow-gpu等等,因为用了清华镜像源,每次都很快就安装好了。不需要再输入-c +channel了,这也是个坑,统一用清华的就行。另外说一下清华的网站不是用来手动下载package的,我一开始下了很多zip或者whl的包,其实并不需要,只要把网址写到channel里面,用conda install就行。

如果需要安装gpu版的pytorch或tensorflow需要显卡是NVIDIA的,先看下自己显卡支持的cuda版本。方法如图5和图6,打开NVIDIA设置->帮助->系统信息->组件->3D设置。我的是1060的显卡,cuda版本是10.2。

Anaconda安装 torch
图5 NVIDIA设置
Anaconda安装 torch
图6 cuda版本

接下来在命令行输入

conda install pytorch torchvision cudatoolkit=10.1

注意命令行一定要是在py3.7-gpu环境下的,如图7。我的显卡是cuda10.2的,因此只要是10.x都行。具体命令可以去pytorch官网看

如图8,注意把最后那个-c pytorch去掉。

Anaconda安装 torch
图7 命令行前缀
Anaconda安装 torch
图8 pytorch配置

全都安装好了可以在命令行查看是否成功,方法为:在命令行先输入

然后输入

最后输入

print(torch.cuda.is_available())

返回True就大功告成了。如果装的是tensorflow-gpu,就输入

以及

tf.test.gpu_device_name()

返回

就成功了。

最后看一下用cpu和gpu跑程序速度的区别,如图9和图10所示。

Anaconda安装 torch
图9 CPU的速度
Anaconda安装 torch
图10 GPU的速度

以后有想到的再补充,侵删~

一.介绍:

由于昨天Windows重装系统了,很多配好的环境都丢掉了

本期分享一个win10下安装python环境,安装anaconda,外加PyTorch的教程,按照教程1小时内可以全部配好相信这是速度最快的方法!同时也是手把手的教程!!

首先下载好最新的Anaconda安装包,

Ctrl+F,输入2020,找到最新的exe,萝卜电脑是win10+64位的,选择自己系统对应的下载,然后可以感受到国内源飞一般的速度,大概5分钟可以下好466MB的安装包

Anaconda安装 torch
Anaconda安装 torch
点击exe
Anaconda安装 torch
选next

依次选择,I agree(我愿意!!),next

Anaconda安装 torch

选择默认路径C盘,如果C盘空间不够的可以安装到其他盘,C盘查找会比较快

需要自己配置好环境变量,参考一下博客,写的非常详细!

之后就安装好了anaconda!

特别注意,anaconda里面是自带Python的,小伙伴萌不需要额外下载Python

在win10的搜索栏输入an,打开anaconda prompt

Anaconda安装 torch

输入python就可以进入anaconda自带的python环境了

Anaconda安装 torch

也可以看看自带python的版本,输入 python --version,可以看到 Python 版本是3.7.6

Anaconda安装 torch
python版本

三.安装pytorch

打开pytorch官网:https://pytorch.org/

Anaconda安装 torch
复制最下面的命令行

YourOS:你的操作系统

Package:选择下载的方式,由于conda要配置源,对萌新不太友好,所以我们选择pip

Language:选择python,(上一次看没咩有支持Java的呀!!??现在可以了吗??!)

CUDA:None是下载CPU版,选择自己系统支持的cuda,这是GPU版

如何查看电脑支持cuda的版本:

最后重点来了!!请做笔记!!直接Run this command会很慢,我们只需要切换到国内源,就会体验飞一般的感觉

只要在命令行后面加上:

-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

就能切换到国内源了,

比如我的就变成了:

pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

在anaconda promt中输入上述命令就行了!再次感受一下飞速

Anaconda安装 torch

安装完成后,验证一下是否成功

打开anaconda prompt,进入python环境,输入import torch

Anaconda安装 torch

成功!!

四.总结

之后就可以愉快的开始使用Torch了!大功告成!

其实最快的方法就是使用国内的源,不管是下载anaconda还是配置库函数

对于初级炼丹师来说,之后需要安装各种库函数,比较推荐的一种方法是:

pip install xxx -i https://pypi.tuna.tsinghua.edu.cn/simple

xxx对应需要安装的库,比如安装numpy就是

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

以上就是本篇教程的全部内容了,觉得有帮助的小伙伴可以点赞关注!!

有什么疑问评论区留言~~