windows TensorFlow-gpu安装

1.安装环境

window10+GTX1060+cuda9.0+cudnn 7.0

2.安装TensorFlow-GPU

pip install tensorflow-gpu

本人安装的是tensorflow-gpu 1.8.0

如果出现网络time out，请按照https://blog.csdn.net/u010349092/article/details/79945891配置国内源

3.安装cuda

注意cuda版本必须和tensorflow-gpu匹配，如我安装了cuda9.2版本，运行工程时报错Could not find 'cudart64_90.dll'，提示需要使用cuda9.0版本

去https://developer.nvidia.com/cuda-downloads下载对应的cuda版本，默认安装即可

4.安装cudnn

同样，cudnn版本也需要配套，如在工程中报错，could not find 'cudnn64_7.dll'，说明需要安装cudnn-9.0-windows10-x64-v7，其中9.0对应的是cuda版本

去https://developer.nvidia.com/cudnn下载对应的cudnn版本，解压

将对应的bin、include、lib目录下的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目录下对应的bin、include、lib目录下

重启notebook，即可运行工程。