移植TensorFlow到Windows平台

2015年11月,Google宣布开源旗下机器学习工具TensorFlow,引发业界热潮。TensorFlow原生支持*unix系和安卓平台,但并不提供对Windows平台的支持。如果想在Windows上运行TensorFlow,官方提供了Docker/Linux虚拟机的绕行方案。

2016年4月,出于工作的安排,我承担了将TensorFlow(当时的版本是0.8)移植到Windows平台的任务。TensorFlow项目的代码包括C++内核和Python API两部分,前者用于科学计算,后者简化DAG构建。在此之前,我一直混迹在JAVA阵营,对于C++和Python的知识局限在大学课本的水平。因此,这个移植任务对我而言还是难度很大的,跨平台的语法、编译、链接、测试、打包等等一切都要靠自己慢慢摸索。

历时两个月,在7月初我独自一人完成了TensorFlow初版的移植,运行情况良好。由于性能有限以及团队缺少资源维护代码的时效性等原因,我的这一产出并没有最终送入生产环境。除了提供给另一团队进行TensorFlow的扩展开发外,基本被搁置了——目前团队集中精力在另一套TensorFlow跨平台的技术方案上。

为了不让我的这份工作浪费,我将自己的产出开源到GitHub上,对机器学习或者跨平台移植感兴趣的同学可以自由的clone到本地并进行编译、打包、安装和使用。

项目地址:https://github.com/shishaochen/TensorFlow-0.8-Win

posted @ 2016-08-23 10:48  玉龙gg  阅读(1782)  评论(0编辑  收藏  举报