windows10 64bit 下的tensorflow 安装

本文参考https://www.cnblogs.com/marszhw/p/6658990.html以及补充以下几点注意事项。

注：执行pip install --upgrade tensorflow-gpu后为最新版本的tensorflow-gpu,所以后面要下载安装对应版本的CUDA库。

安装TensorFlow后，找到C:\Python35\Lib\site-packages\tensorflow\python\platform目录下的build_info.py文件。用编辑器打开可查看CUDA和cuDnn库的版本信息。

2、安装CUDA库

    cuda库下载 链接：https://developer.nvidia.com/cuda-toolkit-archive。安装完成后环境变量默认是配好的。

3、添加cuDnn库

    下载链接：https://developer.nvidia.com/rdp/cudnn-download。解压后，将lib文件夹拷贝到CUDA库的bin目录下，默认安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 。将bin文件夹里的文件拷贝至CUDA库同名bin目录下。将include文件夹里的文件拷贝至CUDA库同名的include目录下。

4、测试TensorFlow

详细demo可参照本文开头参考链接。