深度学习图像分类之下载安装Pytorch

1. 简述Pytorch

PyTorch是一个开源的Python机器学习库,其前身是著名的机器学习库Torch。2017年1月,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTorch,它是一个面向Python语言的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络,这是很多主流深度学习框架比如Tensorflow等都不支持的。PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了Facebook外,它已经被Twitter、CMU和Salesforce等机构采用。作为经典机器学习库Torch的端口,PyTorch 为 Python 语言使用者提供了舒适的深度学习开发选择。

目前Pytorch成为深度学习领域使用最火热的框架其原因主要包括以下三点:
(1) 简洁; (2)速度; (3)易用;

2. 安装步骤:

2.1 安装Python

由于Pytorch面向的是Python语言,因此首先需要安装Python。
由于之前已经安装过Python,因此这里不再重复安装(我安装的版本是3.7)。
Python的安装可以参考:全网最详细的Python安装教程(Windows) - 知乎 (zhihu.com)

2.2 安装CUDA

要特别注意的是Pytorch对CUDA的版本是有要求的,因此我们安装CUDA之前需要先确认到底装哪个CUDA版本才行。
(1) 首先我们要确定本机是否有独立显卡。计算机->设备管理器->显示适配器 中,查看是否有独立显卡,如下图所示:

查看独立显卡

如上图所示,可以看到,当前系统拥有NVIDIA显卡,型号为GeForce GTX 960M。
(2) 其次我们要查看本机显卡适合什么版本的CUDA。NVIDIA控制面板->帮助->系统信息->组件

查看适合的CUDA

由上图可知本机适合的CUDA版本为8.0。
(3) 开始安装CUDA。
根据正常思路,我们进入CUDA安装官网安装:https://developer.nvidia.com/cuda-toolkit-archive,但是我发现该官网无论我用什么方法都无法进入,于是我参考该文章关于下载cuda和cudnn官方网站访问不了的替代方案_sinat_29217765的博客-CSDN博客将网址https://developer.nvidia.com/cuda-toolkit-archive中的com改为cn后(https://developer.nvidia.cn/cuda-toolkit-archive)可顺利访问。

CUDA官网

下面我们选择 CUDA Toolkit 8.0进行下载。单击后会进入版本配置界面,按下图进行选择即可:

下载信息选择

下载完成后便自动进入安装程序。

安装

根据提示一步步安装即可,安装完毕之后打开cmd命令终端测试一下cuda是否安装成功
输入命令测试:

 nvcc -V

正常情况下会出现下图所示结果,说明安装成功:

测试结果

2.3 安装cuDNN。

由于官网下载cuDnn需要加入Nvidia developers会员才能进行下载,因此参考文章官网安装Nvidia CUDA Toolkit、cuDNN报错502的备用镜像下载网址_sinat_28371057的博客-CSDN博客_cudnn 镜像进行下载。要注意的是CUDA8.0对应的cuDNN版本是5.1。
下载之后,解压缩,将CUDNN压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0),直接覆盖安装即可完成。

2.4 安装Pytorch

安装完CUDA和cuDNN后,安装Pytorch相对比较简单。(虽然我还是遇到了蛮多问题)

进入Pytorch的安装选择配置页面,此时根据相关配置会自动给出安装方法,如下所示:Start Locally | PyTorch

Pytorch的安装选择配置页面

但是由于我根据本机的配置安装的版本为CUDA 8.0因此不在此页面内,我参考文章PyTorch 安装(CUDA 是8.0以下旧版本,如何安装 pytorch?)_u014546828的博客-CSDN博客进行安装,只需要在cmd终端中运行相应的命令即可:

pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

但是由于使用cmd终端进行pip下载的时候一直下载到一半卡住,于是参考文章Python:pip install torch官网速度太慢的解决办法_谁偷了我的酒窝的博客-CSDN博客使用迅雷下载相应的安装包。下载完成后,用命令行打开文件所在的路径,重新输入安装命令。

安装torch

安装torchvision的时候我看许多文章表示安装完正确版本的torch后直接pip install就行,于是我照做了,于是我出错了,下面是我踩坑过程,请勿模仿

安装torchvision

从上图我们可以看到,系统发现我之前按照了torch0.4.1于是它把我的0.4.1版本删了删了删了,换成了其他版本...我刚开始没认真看于是接着验证了。

2.5 验证

验证Pytorch是否安装成功并且能够成功调用cuDNN。

验证Pytorch

使用命令行进行验证后发现,虽然安装成功但是安装的是cpu版本,因此导致验证False,问题应该是出在前面安装完torch后采用自动安装torchvision的时候它自动匹配选择了版本,因此我只能卸载重新安装,卸载命令很简单,就是把之前安装时的install改成uninstall即可。

卸载torch与torchvision

经历惨痛经历的我这次我根据文章pythorch版本和torchvision版本对应关系及torchvision安装_村西那条弯弯的河流的博客-CSDN博客下载相应版本的torchvision再进行pip安装。

版本对应

因此下载0.2.2版本的torchvision即可,但是在官网的链接里我并没有找到这个版本的torchvision,于是我在清华源找到了这个版本Links for torchvision (tsinghua.edu.cn)

清华源

找到链接后一样的方法使用迅雷进行下载,速度超快,几秒钟就完事了,然后打开下载路径进行pip安装即可。安装完成后重新进行验证。

验证成功

历经磨难!由图可知,版本正确,并且cuda正常可用,说明我们的Pytorch安装成功告一段落,撒花!

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 162,825评论 4 377
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 68,887评论 2 308
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 112,425评论 0 255
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,801评论 0 224
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 53,252评论 3 299
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 41,089评论 1 226
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 32,216评论 2 322
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 31,005评论 0 215
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,747评论 1 250
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,883评论 2 255
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,354评论 1 265
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,694评论 3 265
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,406评论 3 246
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,222评论 0 9
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,996评论 0 201
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 36,242评论 2 287
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 36,017评论 2 281

推荐阅读更多精彩内容