前提:利用anaconda安装了tensorflow; CUDA 8.0; cudnn 6.0(具体参见github上说明https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)
步骤:
1、激活环境
source activate tensorflow
2、安装需要的tensorflow版本(以gpu版1.3为例子)
pip install -U tensorflow-gpu==1.3
3、检验是否安装成功
进入python;
输入import tensorflow as tf,不报错,则安装成功;报错要看具体错误。我遇到的就是cudnn版本不匹配,然后换个版本就OK了。还有运行代码的过程中会遇到,编译未使用SSE4.1,SSE4.2,AVX,AVX2和FMA,在开头要加上环境说明,是否显示所有信息,warning,error。
参考博客:
查看cuda,cudnn版本https://blog.youkuaiyun.com/Teeyohuang/article/details/79082003
cudnn6.0下载https://www.zhihu.com/question/37082272
cudnn安装https://blog.youkuaiyun.com/enjoyyl/article/details/78086459
编译未使用SSE4.1,SSE4.2,AVX,AVX2和FMA的解决方法https://blog.youkuaiyun.com/tsinghuahui/article/details/72938764