linux – Pycharm tensorflow ImportError但可以正常使用Terminal

前端之家收集整理的这篇文章主要介绍了linux – Pycharm tensorflow ImportError但可以正常使用Terminal前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我使用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的终端中运行没有错误,但只要我点击运行按钮就会给我上面的错误.环境变量还有问题吗?我如何解决它?

解决方法

不幸的是,在pycharm中你必须在IDE中设置CUDA_HOME和LD_LIBRARY_PATH.为此,请转到“运行”菜单 – >编辑配置,然后选择您的项目.然后单击Environment Variables并为CUDA_HOME和LD_LIBRARY_PATH添加一个条目.完成后,您可以直接从pycharm内部运行.我还建议设置默认值以包含这些路径,因此您不必为每个项目执行此操作.另外,当您在此设置页面上时,确保从源[virtualenv_path]设置解释器也没有什么坏处.如果您无法找到执行此操作的位置,请与我们联系.

猜你在找的Linux相关文章