0
点赞
收藏
分享

微信扫一扫

PyTorch简介与安装

静鸡鸡的JC 2022-02-23 阅读 99



文章目录


  • ​​1.什么是PyTorch,为什么选择PyTorch?​​
  • ​​2.PyTorch安装(windows 10)​​


​PyTorch​​​ 是 ​​Torch​​​ 在 ​​Python​​​ 上的衍生(​​Torch​​​ 是一个使用 ​​Lua​​ 语言的神经网络库) 。

和​​tensorflow​​比较


  • ​PyTorch​​建立的神经网络是​动态的​, ​​Tensorflow​​是建立​静态图
  • ​Tensorflow​​​ 的高度工业化, 它的底层代码是很难看懂的. ​​PyTorch​​​ 好那么一点点, 如果你深入 ​​API​​​, 你至少能比看 ​​Tensorflow​​​ 多看懂一点点 ​​PyTorch​​ 的底层在干嘛.

建议:


  • 如果你是学生, 随便选一个学, 或者稍稍偏向​​PyTorch​​​, 因为写代码的时候应该更好理解. 懂了一个模块, 转换 ​​Tensorflow​​或者其他的模块都好说.
  • 如果是上班了, 跟着你公司来, 公司用什么, 你就用什么, 不要脱群.

1.什么是PyTorch,为什么选择PyTorch?

(1)什么是PyTorch?

​PyTorch​​​的前身便是​​Torch​​​,其底层和​​Torch​​​框架一样,但是使用​​Python​​​重新写了很多内容,不仅更加灵活,支持动态图,而且提供了​​Python​​​接口。它是由​​Torch7​​​团队开发,是一个以​​Python​​​优先的深度学习框架,不仅能够实现强大的​​GPU​​​加速,同时还支持动态神经网络,这是很多主流深度学习框架比如​​Tensorflow​​等都不支持的。

​PyTorch​​​既可以看作加入了​​GPU​​​支持的​​numpy​​​,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了​​Facebook​​​外,它已经被​​Twitter​​​、​​CMU​​​和​​Salesforce​​等机构采用。

(2) 为什么选择PyTorch?

面对如此多的深度学习框架,我们为何要选择​​PyTorch​​​ 呢?​​Tensorflow​​​不是深度学习框架默认的老大吗,为什么不直接选择​​Tensorflow​​​而是要选择​​PyTorch​​​ 呢?下面分4个方面来介绍为何要使用​​PyTorch​​。

(1)掌握一个框架并不能一劳永逸,现在深度学习并没有谁拥有绝对的垄断地位,就算是​​Google​​​ 也没有,所以只学习​​Tensorflow​​ 并不够。同时现在的研究者使用各个框架的都有,如果你要去看他们实现的代码,至少也需要了解他们使用的框架,所以多学一个框架,以备不时之需。

(2)​​Tensorflow​​​ 与​​Caffe​​​都是命令式的编程语言,而且是静态的,首先必须构建一个神经网络,然后一次又一次使用同样的结构,如果想要改变网络的结构,就必须从头开始。但是对于​​PyTorch​​​,通过一种反向自动求导的技术,可以让你零延迟地任意改变神经网络的行为,尽管这项技术不是​​PyTorch​​​ 独有,但目前为止它实现是最快的,能够为你任何疯狂想法的实现获得最高的速度和最佳的灵活性,这也是​​PyTorch​​​对比​​Tensorflow​​ 最大的优势。

(3)​​PyTorch​​​ 的设计思路是线性、直观且易于使用的,当你执行一行代码时,它会忠实地执行,并没有异步的世界观,所以当你的代码出现​​Bug​​​ 的时候,可以通过这些信息轻松快捷地找到出错的代码,不会让你在​​Debug​​ 的时候因为错误的指向或者异步和不透明的引擎浪费太多的时间。

(4)​​PyTorch​​​的代码相对于​​Tensorflow​​​而言,更加简洁直观,同时对于​​Tensorflow​​​高度工业化的很难看懂的底层代码,​​PyTorch​​​ 的源代码就要友好得多,更容易看懂。深入​​API​​​,理解​​PyTorch​​底层肯定是一件令人高兴的事。一个底层架构能够看懂的框架,你对其的理解会更深。

2.PyTorch安装(windows 10)

(1)python安装,​ 当前最新的正式版本是​​python3.7.3​​,下载地址:https://www.python.org/ftp/python/3.7.3/python-3.7.3-amd64.exe

(2)安装Anaconda,​ 要注意和​​python​​对应的版本,下载地址:https://www.anaconda.com/download/

附一张anaconda与python版本对应关系

PyTorch简介与安装_深度学习

首先解释一下上表。 ​​anaconda​​在每次发布新版本的时候都会给​​python3​​和​​python2​​都发布一个包,版本号是一样的。

表格中,​​python​​版本号下方的离它最近的​​anaconda​​包就是包含它的版本。

举个例子,假设你想安装​​python2.7.14​​,在表格中找到它,它下方的三个​​anaconda​​包(​​anaconda2-5.0.1、5.1.0、5.2.0​​)都包含​​python2.7.14​​;

假设你想安装​​python3.6.5​​​,在表格中找到它,它下方的​​anaconda3-5.2.0​​就是你需要下载的包;

假设你想安装​​python3.7.0​​,在表格中找到它,它下方的​​anaconda3-5.3.0​​或​​5.3.1​​就是你需要下载的包;

(3)安装cuda(如果没有显卡,省略此步)

(4)运行pytorch的安装命令,从官网可以生成安装命令:

PyTorch简介与安装_python_02



举报

相关推荐

0 条评论