Windows 编译tensorflow1.3-gpu源码

一、环境准备：

vs2015（cuda8.0最新只支持到vs2015，所以vs2017及以上编译不过）

cuda8.0  下载地址

cudnn6.0 下载地址

swig-win 下载地址

cmake

tensorflow-r1.3源码 下载地址

 

二、安装软件

1.安装vs2015

默认安装即可

 

2.安装cuda8.0

默认安装即可

配置环境变量

 

3.安装cudnn6.0

解压cudnn6.0压缩包，把对应文件放到cuda的安装目录下

cudnn64_6.dll 拷贝到cuda8.0的bin/ 目录下

cudnn.h 拷贝到cuda8.0的include/ 目录下

cudnn.lib 拷贝到cuda8.0的lib/x64/目录下

 

 

4.安装cmake

默认安装即可

 

三、生成tensorflow的vs工程

1.打开cmake-gui，选择tensorflow-r1.3的cmake文件夹，并选择build文件夹，依次点击configure->generate->open project

configure时如果报swig.exe找不到，手动选择一下，重新configure即可。

注意：需要手动勾选enable-gpu，我这里不需要grpc，所以把grpc去掉了。

 

2.编译tensorflow-gpu工程

注意：编译的过程中会在线下载依赖的工程源码，所以这里网络一定要好，否则会一直下载失败；如果遇到翻墙的文件，可以用国内的镜像文件替换。当然也可以手动下载好，放在build/download文件夹下。