tensorflow - Pycharm 在从终端运行的脚本上给出错误(模块 : Tensorflow)

Question

我在 Pycharm 中使用 tensorflow(GPU 版本)模块。如果我从终端运行脚本，它会按预期工作。然而，当我从 pycharm 运行脚本时，它说:

ImportError: libcudart.so.7.5: cannot open shared object file: No 
such file or directory

我该如何解决？

Pycharm 解释器将 tensorflow 显示为一个包。

在终端中，当我检查tensorflow的版本时，它与pycharm(0.10.0rc0)中的版本相同

Answer 1

看起来您的 CUDA_HOME 或 LD_LIBRARY_PATH 在控制台中配置正确，但在 PyCharm 中配置不正确。您可以在控制台中检查和比较它们的值

echo $CUDA_HOME
echo $LD_LIBRARY_PATH

在 PyCharm 中(比如，在您的主脚本中):

import os
print(os.environ.get('CUDA_HOME'))
print(os.environ.get('LD_LIBRARY_PATH'))

您可以在环境变量部分为给定的运行配置配置它们。

更好的方法是全局配置这些环境变量，这样系统中的每个进程都可以访问它们。为此，您必须编辑 /etc/environment 文件并添加您从控制台获得的原始值。

这里有非常相似的问题:one , two , three .