tensorflow 在windows 下怎么使用gpu

需要安装的工具

  • python 3.5.2
  • tensorflow-gpu (0.12.1)
  • Cuda Toolkit 8.0
  • cuDNN

安装tensorflow-gpu

1
pip install tensorflow-gpu

安装Cuda Toolkit 8.0

Cuda Toolkit 8.0下载地址

Cuda Toolkit 8.0
Cuda Toolkit 8.0

安装前注意更新显示驱动! 下载后安装一路下一步

安装cuDNN

cuDNN
cuDNN
  • 解压

  • 添加bin目录到环境变量PATH

安装cuDNN
安装cuDNN

测试

测试1
测试1
测试2
测试2
测试3
测试3

注意1: 如果都安装好但tensorflow没反应的话, 先卸载tensorflow tensorflow-gpu 然后重装

注意2: 运行报错Couldn’t open CUDA library cupti64_80.dll 解决方法:${CUDA_HOME}/extras/CUPTI/lib64 添加到PATH环境变量