笔者在win10系统下安装tensorflow也是吃尽了苦头,折腾了许久,参考了很多博客与教程,终于安装成功,分享一下我遇到的问题,希望能造福各位。
Attention:以下所有内容仅适配 win10 64位系统+Tensorflow 1.4.0
1、安装Cuda 8.0
首先,不同版本的Tensorflow对应的cuda tookit和cudnn是不同的,所以大家要对应自己安装的tensorflow版本去下载安装这两个工具。
笔者安装的是tensorflow1.4.0,对应的cuda是cuda_8.0.61,此文件可以在官网地址下载,按图选择
network和local的区别是,一个在线安装,一个本地安装包,随便选。
选好后选择第一个进行下载
下载完成一路next安装就好,记得点选添加path,这样就不用手动再配一遍环境变量
测试是否安装成功:
打开命令行cmd,输入nvcc -V
输出版本信息则安装成功
2、安装cudnn6.0
cudnn的官网下载流程非常麻烦,需要你先填写问卷,成为NVIDIA的开发者,等不及如我,直接百度网盘搞定。
我安装的是cudnn6.0,适配cuda8.0,下载好压缩包名称是cudnn-8.0-windows10-x64-v6.0,大家不要搞错了。
下载完,解压,把解压好的文件,放入刚刚cuda对应的目录下,bin对bin,lib对lib,大地对长空,山花对海树,and so on