笔者因为想尝试一些机器学习方面的idea,所以于TensorFlow产生了交集, 笔者搞计算机图形学,所以更多地与windows和visual studio打交道, 于是想在windows和visual studio环境下编译出tensorflow的gpu版本。
但是整个互联网对于在windows和vs2015下编译tensorflow的信息少的可怜, 甚至在tensorflow的官方git hub页面, 也宣称没有在windows+vs环境下成功build出tensorflow GPU。
所以, 似乎, 为了搞点儿机器学习的算法,我们就不得不放弃可爱的windows屈从于Linux了? -- 就因为那些不搞游戏编程的人喜欢用Linux?不可能的。
在开始这篇文章前, 我首先要讲一下为什么要编译tf源码。(从此以后tf就是tensorflow的简称)。
为此, 我首先要声明,如果读者不想使用tensorflow的高级功能的话, 在windows上安装它的python接口就可以了, 很简单, 也很方便, 同样是带了gpu支持的,是tensorflow开发者对于windows python预编好的接口, 对此, 有以下几点需要注意:必须先安装好cuda8.0, 其它版本不建议, cuda8.0可从nvidia的官网获得, 使用默认安装路径即可
之后安装cudnn5.0, 任何其它版本不适用。 因为nvidia提供的cudnn只是一个 .zip文件, 所以解压后把里头的 \bin; \include; \lib; 文件夹合并到cuda的安装目录下, 比如我的为: C:\Program Files\NVIDIA GPU Computing Toolki