Why choose pytorch
在深度学习领域,一般用于开发的主流深度学习框架包括:Caffe, TensorFlow, MXNet, pytorch。除此之外国内还有百度的paddpaddle等。笔者只用过Caffe,浅浅了解TensorFlow。 Caffe是一个优秀开源深度学习框架,但是目前早已经停止更新,而且caffe使用了许多第三方库,每次安装编译caffe都不是一帆风顺。TensorFlow只是简单了解,有许多自创的概念比如Session, Variable等, 写起来很啰嗦。
pytorch是Facebook的开源深度学习框架,背后有大公司的支持,而且有活跃的pytorch社区,文档也很齐全,对于开发者而言很友好。
开发环境
- Ubuntu 18.04
- Anaconda3
- pycharm
pytorch 安装
首先进入pytorch官网:
根据自己电脑的环境,选择不同的版本:
因为本人开发环境为Ubuntu, Anaconda3, python3.6因此在终端执行: conda install pytorch torchvision -c pytorch
等待数分钟就自动安装完毕。
在终端输入:conda list
在终端进入python,输入:
import torch
import torchvision
说明pytorch, torvision安装成功。