Win10下配置TensorFlow-GPU

环境:
Win10
Anaconda3(4.2)、Pyhon3.5、tensorflow-gpu1.4
Cuda8.0、cudnn6.0

首先是需要的一些软件包,包括cuda、anaconda、cudnn等,我直接放在了百度云,方便各位看官的下载~

链接: https://pan.baidu.com/s/19PVVE6AdPYA9dRaoSdOUHQ 密码: rykz

首先将百度云里面的文件下载,备用

1、安装Cuda

安装cuda_8.0.61_win10.exe文件,下一步下一步,正常安装

2、配置Cuda环境

安装完成后将CUDA安装目录下的binlibnvpp配置到系统Path环境变量中,或者你打开Path之后看到这个路径已经存在,那也是可以的

3、cuDNN

将cuDNN的压缩包cudnn-8.0-windows10-x64-v6.0.zip解压,得到的三个文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0中(根据自己的安装路径,基本上就是这个)

4、Anaconda3-4.2(对应的Python3.5)

同样的,安装百度云下载的Anaconda3-4.2.0-Windows-x86_64.exe,exe文件安装,so easy!

5、安装TensorFlow-GPU

右键Win10左下角的开始,选择命令提示符(管理员),输入pip install tensorflow-gpu==1.4.0,对应版本我用的是1.4的,最新的版本一般会出错误~

OK,没啦,可以验证一下
命令行里面

python
import tensorflow as tf