尝试使用 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

如果您在 GPU 上使用 TensorFlow，则需要安装 CUDA 和 cuDNN。请按照https://www.tensorflow.org/install/ 上的说明进行操作

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