[英]“Failed to load the native TensorFlow runtime.” error when importing tensorflow with rocm
[英]How to fix "Failed to load the native TensorFlow runtime." error when attempting running on gpu?
我一直在我的 CPU 上運行我的 tensorflow 程序。 然而,它變得越來越慢,所以我想設置一個gpu環境在我的1070ti上運行它。 但是,在下載 CUDA 9.0 和 CUDNN 7.4 后,tensorflow-gpu 無法運行。
每當我嘗試在 python 中導入 tensorflow 時都會出現錯誤。 我在 virtualenv 環境中安裝了 tensorflow,並且運行的腳本在文件夾內。
import tensorflow
我得到以下輸出:
Traceback (most recent call last):
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\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 "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\imp.py", line 243, in load_module
return load_dynamic(name, filename, file)
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\imp.py", line 343, in load_dynamic
return _load(spec)
ImportError: DLL load failed: The specified module could not be found.
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 74, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\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 "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\imp.py", line 243, in load_module
return load_dynamic(name, filename, file)
File "C:\Users\Jimmy\Documents\machinelearning\machinelearning\lib\imp.py", line 343, in load_dynamic
return _load(spec)
ImportError: DLL load failed: The specified module could not be found.
根據這個線程,這些包在 Windows 10 上不起作用:
但是,有人說可以嘗試使用 pip 將 Tensorflow 降級到 1.10 版本:
pip install tensorflow-gpu==1.10.0
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.