一直以来绝大多数的深度学习应用都是基于Keras和Tensorflow搭建的,很少有去花时间使用过Pytorch,最近正好有时间就想学习下这个被称为最适合做学术研究的深度学习框架。
这里主要是记录下自己windows环境下的环境搭建过程,希望能帮到需要的人。
安装Pytorch的方式比较特殊,虽然也是基于pip安装的,但是又不能像安装Keras一样,直接pip安装,这样的话会报错要你参考官网给出来的安装方式。
先来看下Pytorch的官网,在这里,首页截图如下所示:
这里我们选择的是本地安装模式【Locally】,按照对应的选项点击选择完成之后,对应的安装命令就已经出现在最下方的文本框里面了,比如这里对于我CPU的安装方式来说,安装Pytorch的命令如下:
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
复制到CMD窗口里面,直接执行就好了,成功安装后,导入使用截图如下所示:
到此,确定安装成功,后面可以学习使用了,记录一下。