TensorFlowCXX编译与配置
一、所需软件工具
1.CMake version 3.5以上
2.git (http://git-scm.com)
3.swig (http://www.swig.org/download.html)解压即可
4.Visual Studio 2015
5.UCDA9.0 (https://developer.nvidia.com/cuda-90-download-archive)
注意:在2018.08.24是编译的tensorFlow版本时使用ucda9.0
6.CUDNN一定要对应上面的UCDA9.0版(https://developer.nvidia.com/cudnn)下载后解压合并到安装的UCDA/v9.0目录下。
7.Python 3.5 64-bit (选择添加安装lib文件)
8.NumPy 1.11.0 以上
注意事项:CMake 与 git安装时需要将其加入到环境变量中,否则后续编译过程中会出现找不到相关程序的错误
二、以上所需工具全部安装全后,在guit拉下TensorFlow的源码:
https://github.com/tensorflow/tensorflow.git
git clone https://github.com/tensorflow/tensorflow
cd tensorflow
#创建这个build目录
mkdir build
三、注意在第一步时一定要把所有配置好才能进行下一步操作
1、首先启用vs2015 x64编译否则很容易堆空间不足的错误提示。
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\amd64\vcvarsall.bat
2、cuda和cudnn的路径设置,一定要设置好否则不能通过gpu的编译。在系统设置PATH中添加如下。其实只添加到v9.0这个根目录就可以。为了保险起见多添加了两个。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.