尝试使用gpu导入tensorflow时导入错误

Question

ImportError: libcuda.so.1: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.

导入tensorflow时出现此错误。 我需要知道解决这个问题的步骤。

Answer 1

如果您在TensorFlow中使用GPU，则需要安装CUDA和cuDNN。请按照https://www.tensorflow.org/install/

上的说明操作

如果您已经安装了CUDA和cuDNN，但仍然出现此错误，那么您可能忘记导出库：对于Linux，您可能需要设置LD_LIBRARY_PATH以包含CUDA库。

Answer 2

在通过Anaconda安装了TensorFlow 1.11 GPU之后，“ conda install tensorflow-gpu”也遇到了同样的问题。在TF 1.11之前，我使用TF 1.04。在TF更新之前，我通过ppa将Nvidia驱动程序更新为396版。

在我的整个Linux中都没有libcuda.so.1的痕迹。

经过多次试验，当我将nvidia驱动程序更改为390时，问题得以解决。390版本将libcuda.so.1插入/ usr / lib / i386-linux-gnu目录，这显然解决了问题。