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安装目录下的bin
和libnvpp
配置到系统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