pytorch 1.0 一个月之前发布了。pytorch其实笔者很早就接触过,那时候惊叹于它的简洁、动态及良好的社区支持。但是那时候,pytorch在c++上的支持并不好,工业界很难用,基本上只属于一种比较好的算法验证框架。
但是,pytorch 1.0 的发布终结了这一现况,现在,它有了良好的c++前向预测支持,再加上它本身不逊于tensorflow的运行速度,python式的开发逻辑(这点非常棒),支持onnx格式。。。除了没有tensorflow那样多的功能函数(这点也造成了tensorflow的冗长),实在是很难找到别的缺点。
为什么要自己编译pytorch
pytorch的官方网站上,其实有编好的pytorch各种版本(windows/linux/Mac环境下,python/c++两种语言,从这也看出pytorch对windows的用户还是挺友好的),但是目前c++版本好像只支持release模式。如果你是一个初学者,可以选择跳过本文直接去官网下;但如果你是一个研究者,需要变动源码的contributor,或和我一样,需要Debug版本和Cpu版本,或者特定CUDA版本的pytorch,你也可以选择这条更为艰辛但收获更多的路。
需要准备
1.python(最好是anaconda)
2.CUDA \ CUDNN(如果你需要GPU版本)
3.VS 2017最新版
4.GIT工具
5.cmake工具
6.intel MKL(如果没有的话,pytorch默认选择用eigen)

最低0.47元/天 解锁文章
2179

被折叠的 条评论
为什么被折叠?



