安装完tensorflow后,使用python导入发现以下的错误
百度之后发现很多网友说是 tensorflow安装的版本过高和cuda不匹配,这一有一个 tensorflow和cuda、cudnn的一个对应版本链接可以参考,https://blog.csdn.net/omodao1/article/details/83241074
查看本人环境下的cuda版本和cudnn的版本
1、查看cuda版本
cat /usr/local/cuda/version.txt
2、查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
发现cuda版本为9 cudnn版本为7 符合要求
解决方法:
原因:路径未加入环境
vim ~/.bashrc 在文档尾部加上export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64/
source ~/.banshrc
然后tensorflow就可以用了