TensorFlow 和 Pycharm

Posted

技术标签:

【中文标题】TensorFlow 和 Pycharm【英文标题】:Tensorflow and Pycharm 【发布时间】:2017-10-16 15:11:03 【问题描述】:

我对 pycharm 上的 tensorflow 有疑问。

每当我在 linux 终端中导入 tensorflow 时,它都能正常工作。但是,在 PyCharm 社区 2017.1 中,它显示:

ImportError: libcudnn.so.5: 无法打开共享对象文件:没有这样的文件或目录

关于如何解决问题的任何提示。

请注意,我使用的是 python 3.5.2、tensorflow 1.1.0、Cuda 8 和 CuDnn 5.1

编辑:打印 sys.path 时,我在 PyCharm 中得到了这个:

['/home/xxx/pycharm-community-2017.1.2/helpers/pydev', '/home/xxx/pycharm-community-2017.1.2/helpers/pydev', '/usr/lib/python35. zip', '/usr/lib/python3.5', '/usr/lib/python3.5/plat-x86_64-linux-gnu', '/usr/lib/python3.5/lib-dynload', '/ usr/local/lib/python3.5/dist-packages','/usr/lib/python3/dist-packages','/usr/local/lib/python3.5/dist-packages/IPython/extensions',' /home/xxx/xxx/xxx']

这在终端: ['', '/usr/local/bin', '/usr/lib/python35.zip', '/usr/lib/python3.5', '/usr/lib/python3.5/plat-x86_64-linux -gnu','/usr/lib/python3.5/lib-dynload','/usr/local/lib/python3.5/dist-packages','/usr/lib/python3/dist-packages',' /usr/local/lib/python3.5/dist-packages/IPython/extensions', '/home/xxx/.ipython']

【问题讨论】:

你能检查sys.path在通过终端和通过PyCharm运行时是否匹配? 我刚刚编辑了我的问题。 您能否尝试将此代码添加到脚本的开头(在其他导入之前)并通过 PyCharm 运行:import sys; sys.path.extend(["/usr/local/bin", "/home/xxx/.ipython"])?问题还在吗? @Mirac7 谢谢,但问题仍然存在。 【参考方案1】:

解决办法是:

从控制台运行 PyCharm。 或 将环境变量添加到 IDE 设置中:LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

【讨论】:

以上是关于TensorFlow 和 Pycharm的主要内容,如果未能解决你的问题,请参考以下文章

AI常用框架和工具丨9. 深度学习框架TensorFlow

Tensorflow教程Tensorflow的下载和安装

如何用tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件

tensorflow记录训练和验证的loss

Tensorflow与Keras区别

关于tensorflow的显存占用问题