Tensorflow1.0升级到1.3以上

前提:利用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.csdn.net/Teeyohuang/article/details/79082003

cudnn6.0下载https://www.zhihu.com/question/37082272

cudnn安装https://blog.csdn.net/enjoyyl/article/details/78086459

编译未使用SSE4.1,SSE4.2,AVX,AVX2和FMA的解决方法https://blog.csdn.net/tsinghuahui/article/details/72938764



你可能感兴趣的:(tensorflow)