Pycharm tensorflow ImportError但可以正常使用Terminal

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你必须设置CUDA_HOME和LD_LIBRARY_PATH在IDE中.为此,请转到"运行"菜单 - >"编辑配置",然后选择项目.然后单击环境变量并为CUDA_HOME和添加条目LD_LIBRARY_PATH.完成后,您可以直接从pycharm内部运行.我还建议设置默认值以包含这些路径,因此您不必为每个项目执行此操作.source [virtualenv_path]当你在这个设置页面上时,确保你有来自set 的解释器也没有什么坏处.如果您无法找到执行此操作的位置,请与我们联系.