无法正确导入具有GPU支持的Tensoflow

Question

我的python版本是2.7.12

cuda版本是9.0

cudnn版本是7.0

而我的操作系统是ubuntu 16.04

但是当我导入tensorflow时，我收到以下错误消息。

我该怎么解决！

ImportError：libcusolver.so.8.0：无法打开共享对象文件：没有此类文件或目录

无法加载本地TensorFlow运行时。

请参阅https://www.tensorflow.org/install/install_sources#common_installation_problems

出于某些常见原因和解决方案。 寻求帮助时，在此错误消息上方包括整个堆栈跟踪。

Answer 1

我以前有这些问题。 请指定您要安装的tensorflow版本。 您可以尝试以下解决方案：

解决方案1-

1. 检查您的〜/ .bashrc文件中是否设置了Cuda路径，应该类似于以下export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64\\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

2. 您应该能够运行nvcc -V

3. 如果某个时候一切看起来正确，您可以通过将cuda-9.0更改为路径中的cuda来解决问题。 奇怪，但是对我有用。

解决方案2-

1. 只需为cuda / lib64文件夹中的文件创建符号链接
2. 由于已经安装了cuda 9.0，因此您的lib64文件夹应具有libcusolver.so.9.0检查。
3. 如果一切看起来正确，请使用ln -s /usr/local/cuda/lib64/libcusolver.so.9.0 /usr/local/cuda/lib64/libcusolver.so.8.0
4. 您将必须手动处理7-8个文件。

试试看，或者报告错误。

干杯