无法使用 PyCharm 和 venv 导入 TF

Question

我读了这个， 这个和这个。 这些不是我的情况。 我也试图用这个来解决我的问题。

问题描述

我安装了 cuda 10.0、pip 19.0.2、python 3.6.7、ipython 7.2.0、tensorflow-gpu 1.13.0-rc1、PyCharm CE 2018.3.4。 所有python模块都安装在虚拟环境中。

我在 PyCharm 中选择了 venv 解释器作为项目解释器和 Python 控制台解释器。

当我尝试导入 tensorflow 时

import tensorflow as tf

我使用终端 ipython 成功，使用 PyCharm 失败。

/home/roman/dev/venv/bin/python /home/roman/dev/box.neuro/test.py
Traceback (most recent call last):
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/home/roman/dev/venv/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/roman/dev/venv/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/roman/dev/box.neuro/test.py", line 1, in <module>
    import tensorflow as tf
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 74, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/roman/dev/venv/lib/python3.6/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/home/roman/dev/venv/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/home/roman/dev/venv/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory


Failed to load the native TensorFlow runtime.

See https://www.tensorflow.org/install/errors

for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.

Process finished with exit code 1

附加信息

我确认我以下列方式使用了相同的 ipython。 我检查了import requests ，注意到它没有在终端和 PyCharm 中导入，将它安装在终端中并验证我现在可以在两者中导入它。

我检查了以下内容：

import os
print(os.environ['PATH'])

注意到终端和 PyCharm 的区别。 然后我在设置->构建、执行、部署->控制台->Python控制台中添加了环境变量PATH 。 现在输出是一样的，但我仍然无法在 PyCharm 中导入 tensorflow。

我检查了以下内容：

import os
print(os.environ['LD_LIBRARY_PATH'])

注意到终端和 PyCharm 的区别。 PyCharm IPython 输出为空，终端 IPython 输出如下：

'/usr/local/cuda-10.0/lib64'

Answer 1

我自己找到了解决方法。 我在设置->构建、执行、部署->控制台->Python控制台中添加了环境变量“LD_LIBRARY_PATH”。 为了使用运行/调试配置，我为它们添加了相同的环境变量。 它不是用户友好的，因为我应该一次又一次地为每个配置做它。

以前我在不同的计算机上都安装了 PyCharm，它可以与 TF 一起正常工作，或者说 cuda 库更好。

另外，我在 PyCharm 跟踪器上发现了问题。 因此，如果您对修复感兴趣，请随时为它投票。