可能原因,CUDA版本不对
1、查看下版本信息
代码语言:javascript复制cat /usr/local/cuda/version.txt
2、根据CUDA和tensorflow版本对应的关系表,检查自己的版本是否匹配
3、如果不匹配
请安装相应的TensorFlow或者CUDA版本
安装对应版本TensorFlow的GPU版本
代码语言:javascript复制pip install tensorflow-gpu==版本号
安装对应版本的CUDA
参考:https://blog.csdn.net/wf19930209/article/details/81879514
4、如果匹配,仍然报错
那是因为你的cuda环境变量配置有误,请执行:
代码语言:javascript复制export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/【CUDA版本】/lib64
例如
代码语言:javascript复制export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64