【问题标题】:"Failed to load the native TensorFlow runtime." error when importing tensorflow with rocm“无法加载本机 TensorFlow 运行时。”使用 rocm 导入 tensorflow 时出错
【发布时间】:2020-05-05 02:09:50
【问题描述】:

我有兴趣使用 pycharm 学习 tensorflow,以用于一些使用 amd 系统(r5 3600x、rx 5700xt)的项目。在意识到如果我安装了 linux,我只会让它与我的 gpu 一起工作,我安装了 ubuntu 18.04 以使用 Windows 10 进行双引导,因为它看起来是 rocm 最受支持的。我按照这里的教程进行操作,只是我没有制作虚拟环境。 https://www.videogames.ai/Install-ROCM-Machine-Learning-AMD-GPU 输入sudo apt install rocm-libs miopen-hip cxlactivitylogger时说找不到包cxlactivitylogger,在cmets中作者说他会研究它但没有必要安装tensorflow,所以我忽略了它。然后我打开了 pycharm,tensorflow 包没有检测到,所以我在 pycharm 中再次安装了 tensorflow-rocm 以确保,但是每当我尝试在 pycharm 或终端中导入它时,它都会给我这个错误。

    Traceback (most recent call last):
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/usr/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/usr/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: librccl.so.1: 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/waiai/PycharmProjects/app/main.py", line 1, in <module>
    import tensorflow
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow/__init__.py", line 101, in <module>
    from tensorflow_core import *
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/__init__.py", line 40, in <module>
    from tensorflow.python.tools import module_util as _module_util
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow/__init__.py", line 50, in __getattr__
    module = self._load()
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow/__init__.py", line 44, in _load
    module = _importlib.import_module(self.__name__)
  File "/usr/lib/python3.6/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow.py", line 74, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/home/waiai/PycharmProjects/app/venv/lib/python3.6/site-packages/tensorflow_core/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/usr/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/usr/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: librccl.so.1: 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.

我对linux没有经验,所以这一切让我很头疼,在此之前我曾尝试安装pytorch,但我遇到了类似的错误,所以我决定尝试tensorflow。如果有人能打扰我的手,将不胜感激,在此先感谢。

【问题讨论】:

    标签: python tensorflow pycharm


    【解决方案1】:

    已修复,我用 rccl 安装了

    sudo apt install rccl
    

    【讨论】:

      猜你喜欢
      • 2017-10-19
      • 1970-01-01
      • 2017-07-17
      • 1970-01-01
      • 2020-07-15
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-04-28
      相关资源
      最近更新 更多