Блог

Главная - Вопросы по программированию - Команда Tensorflow tf.test.is_gpu_available() возвращает False

Команда Tensorflow tf.test.is_gpu_available() возвращает False

#python-3.x #tensorflow

#python-3.x #тензорный поток

Вопрос:

Чего мне не хватает, чтобы заставить tf.test.is_gpu_available() возвращать True? Есть ли какой-то файл конфигурации, который нуждается в обновлении?

Я использую Python 3.7.{1,2}

после: pip install --user tensorflow-gpu==2.0.0-alpha0 (до: pip install --user tensorflow==2.0.0-alpha0 )

неполный список элементов:

 tabulate                 0.8.3
tb-nightly               1.14.0a20190301
tensorboard              1.13.1
tensorflow-datasets      1.0.1
tensorflow-estimator     1.13.0
tensorflow-gpu           2.0.0a0
tensorflow-metadata      0.13.0
termcolor                1.1.0
terminado                0.8.1
terminaltables           3.1.0
testpath                 0.4.2
tf-estimator-nightly     1.14.0.dev2019030115
tqdm                     4.31.1
traitlets                4.3.2
urllib3                  1.24.1
virtualenv               16.0.0

выполнить: import tensorflow as tf

возвращает: модель DLL не найдена с трассировкой стека как:

 C:UsersstephPycharmProjectsImage1TF2>python
Python 3.7.2 (tags/v3.7.2:9a3ffc0492, Dec 23 2018, 23:09:28) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
Traceback (most recent call last):
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:Program FilesPython37libimp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "C:Program FilesPython37libimp.py", line 342, in load_dynamic
    return _load(spec)
ImportError: DLL load failed: The specified module could not be found.

Во время обработки вышеупомянутого исключения возникло другое исключение:

 Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflow__init__.py", line 27, in <module>
    from tensorflow._api.v2 import audio
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflow_apiv2audio__init__.py", line 8, in <module>
    from tensorflow.python.ops.gen_audio_ops import decode_wav
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpython__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow.py", line 74, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:UsersstephAppDataRoamingPythonPython37site-packagestensorflowpythonpywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:Program FilesPython37libimp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "C:Program FilesPython37libimp.py", line 342, in load_dynamic
    return _load(spec)
ImportError: DLL load failed: The specified module could not be found.

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.

затем после: pip install --user tensorflow==2.0.0-alpha0

частичный возврат списка элементов:

 tabulate                 0.8.3
tb-nightly               1.14.0a20190301
tensorboard              1.13.1
tensorflow               2.0.0a0
tensorflow-datasets      1.0.1
tensorflow-estimator     1.13.0
tensorflow-gpu           2.0.0a0
tensorflow-metadata      0.13.0
termcolor                1.1.0
terminado                0.8.1
terminaltables           3.1.0
testpath                 0.4.2
tf-estimator-nightly     1.14.0.dev2019030115

выполнить: print(tf.test.is_gpu_available())

ВОЗВРАТ:

 C:UsersstephPycharmProjectsImage1TF2>python
Python 3.7.2 (tags/v3.7.2:9a3ffc0492, Dec 23 2018, 23:09:28) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> print(tf.test.is_gpu_available())
2019-04-04 12:04:46.612501: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
False

Моя переменная среды PATH равна:

 C:UsersstephPycharmProjectsImage1TF2>path
PATH=C:toolscudalibx64;C:toolscudabin;C:UsersstephAppDataRoamingPythonPython37Scripts;C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1include;C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1extrasCUPTIlib64;C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1libx64;C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1bin;C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1libnvvp;C:Program FilesPython37Scripts;C:Program FilesPython37;C:Windowssystem32;C:Windows;C:WindowsSystem32Wbem;C:WindowsSystem32WindowsPowerShellv1.0;C:WindowsSystem32OpenSSH;C:Program FilesMATLABR2019abin;C:Program FilesMATLABR2018bbin;C:Program FilesMATLABR2018abin;C:Program FilesMATLABR2014abin;C:Program Filesdotnet;C:Program FilesMATLABMATLAB Runtimev901runtimewin64;C:Program Files (x86)NVIDIA CorporationPhysXCommon;C:Program FilesNVIDIA CorporationNVIDIA NvDLISR;C:Program FilesNVIDIA CorporationNsight Compute 2019.1;C:Program FilesIntelWiFibin;C:Program FilesCommon FilesIntelWirelessCommon;C:UsersstephAppDataLocalMicrosoftWindowsApps;;C:Program FilesJetBrainsPyCharm Community Edition 2018.3.5bin;;C:UsersstephAppDataLocalProgramsMicrosoft VS Codebin;C:Program FilesIntelWiFibin;C:Program FilesCommon FilesIntelWirelessCommon

Где c:toolscudabin содержатся cuDNN.DLL

Чего мне не хватает, чтобы заставить tf.test.is_gpu_available() возвращать True? Есть ли какой-то файл конфигурации, который нуждается в обновлении?

Ответ №1:

Я столкнулся с аналогичной проблемой, и решение для меня состояло в том, чтобы сопоставить версию CUDA с версией Tensorflow в соответствии с приведенной ниже таблицей. (т. Е. Tensorflow 1.14 требовал cuda 10.0)

https://www.tensorflow.org/install/source#linux