之前尝试过Tenano,Caffe,Keras,Tensorflow,这几个我个人最喜欢的还是Tensorflow,因为是Google出品,技术崇拜,文档很全,并且我看的DeepMind的paper都是用的Tensorflow框架。之前看的目标检测的代码一般用Caffe的很多,人脸识别方面也是Caffe是主流,但是在调试和使用过程中实在接受不能,总是遇到配置的问题纠结好几天,还要涉及python和matlab的接口连接,感觉很复杂,所以后来变成了tf死忠(汗。
最近看到PyTorch的呼声很高,甚至评论说比tf还要好,所以试一下!
话不多说,首先来安装。
PyTorch的官网(http://pytorch.org/)很简洁,很漂亮,也很直观。安装方法的界面也很赞,与其他的Python库一样,直接pip或者conda就好。只是有点大,有482MB,国内下载起来也很慢,断了好几次,下了好久,所以感觉还是出国做科研靠谱啊。
(未完待续)