Я установил версию Tensorflow GPU и CUDA 9.1 и Python 3.6.5.Однако после того, как я импортировал tenorflow-gpu, я получил следующую ошибку:

Question

Traceback (последний вызов был последним):

  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 75, in preload_check
    ctypes.WinDLL(build_info.cudart_dll_name)
  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\ctypes\__init__.py", line 348, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] The specified module could not be found

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

Traceback (most recent call last):
  File "C:/Users/ranji/AppData/Local/Programs/Python/Python36/ts1.py", line 1, in <module>
    import tensorflow as tf
  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
    self_check.preload_check()
  File "C:\Users\ranji\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 82, in preload_check
    % (build_info.cudart_dll_name, build_info.cuda_version_number))
ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit

Answer 1

В настоящее время Tensorflow поддерживает только CUDA 9.0

Но вы установили CUDA 9.1.Теперь у вас есть две опции:

• Опция 1: вы компилируете Tensorflow из источника, после чего включается поддержка вашей версии CUDA 9.1

• Опция 2: Вы устанавливаете CUDA 9.0 и ждете, пока Tensorflow не поддерживает CUDA 9.1

Я прошел через ту же боль и решил установить CUDA 9.0.Это, вероятно, меньше хлопот.

Также обратитесь к этой проблеме на GitHub относительно текущего статуса поддержки Tensorflow для CUDA 9.1 и как скомпилировать Tensorflow из источника.