linux - Pycharm tensorflow ImportError 但适用于终端

Question

我制作了一个安装了 tensorflow 的 virtualenv，并将 Python 解释器设置更改为 virtualenv 所在的位置。当我运行程序时，它给出了错误:

ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory

我的 .bashrc 文件中写有以下几行

export CUDA_HOME=/usr/local/cuda-8.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:$LD_LIBRARY_PATH
export PATH=${CUDA_HOME}/bin:${PATH}

此外，我在以下路径中找到了 libcudnn.so.5 文件:

/usr/local/cuda-8.0/lib64 

我在打开 PyCharm 之前遇到了这个错误，但我确实设法安装了正确的 cudnn 版本。问题是，我可以使用

$ source [virtualenv_path]

激活 virtualenv 并在终端中运行 python，一切正常，没有显示错误。该程序在 Pycharm 的终端中也运行无误，但每当我单击运行按钮时，它只会给我上述错误。环境变量还会有问题吗？我该如何解决？

Answer 1

不幸的是，在 pycharm 中，您必须在 IDE 中设置 CUDA_HOME 和 LD_LIBRARY_PATH。为此，请转到“运行”菜单->“编辑配置”，然后选择您的项目。然后单击环境变量并为 CUDA_HOME 和 LD_LIBRARY_PATH 添加条目。完成后，您可以直接从 pycharm 内部运行。我还建议将默认设置也设置为具有这些路径，这样您就不必为每个项目都这样做。当您在此设置页面上时，确保您从 source [virtualenv_path] 设置了解释器也没有什么坏处。如果您找不到执行此操作的位置，请告诉我。