tensorflow gpu版の設定がうまくいかない

Question

ディープラーニング用にgpuを購入したので、tensorflowのgpu版を導入しようとしているのですが、エラーが出て使う事が出来ません

http://qiita.com/binzume/items/7a0170fe834e250ddef9
主に此方のサイトを中心にして進めさせていただきました

まずcuda8.0をインストールします
↓
次に、cuDNN v6.0 Library for Windows 10のfor cuda8.0をインストールします
↓
次にcuda8.0の中に入ってあるフォルダbinを環境変数に設定します
具体的なディレクトリは以下になります
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
↓
その後cuDNNのbinフォルダに入ってあったcudnn64_6.dllを上記のディレクトリに入れます
↓
そしてtensoflow-gpuをpipでインストールします

しかしimportすると次のようなエラーが起きます

Traceback (most recent call last):
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
    return importlib.import_module(mname)
  File "C:\python3.5.2\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 986, in _gcd_import
  File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
  File "<frozen importlib._bootstrap>", line 577, in module_from_spec
  File "<frozen importlib._bootstrap_external>", line 906, in create_module
  File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
ImportError: DLL load failed: 指定されたモジュールが見つかりません。

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
    return importlib.import_module('_pywrap_tensorflow_internal')
  File "C:\python3.5.2\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
ImportError: No module named '_pywrap_tensorflow_internal'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\python3.5.2\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
    from tensorflow.python import *
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 52, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
    return importlib.import_module(mname)
  File "C:\python3.5.2\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 986, in _gcd_import
  File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
  File "<frozen importlib._bootstrap>", line 577, in module_from_spec
  File "<frozen importlib._bootstrap_external>", line 906, in create_module
  File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
ImportError: DLL load failed: 指定されたモジュールが見つかりません。

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\python3.5.2\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
    return importlib.import_module('_pywrap_tensorflow_internal')
  File "C:\python3.5.2\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
ImportError: No module named '_pywrap_tensorflow_internal'


Failed to load the native TensorFlow runtime.

See https://www.tensorflow.org/install/install_sources#common_installation_problems

for some common reasons and solutions.  Include the entire stack trace

DLL load failedが出た時は、Visual C++ 2015 redistributableのインストールが必要かもしれないとの事ですが既にインストールされており、また再インストールしてみても同じ結果でした

その他pythonやcudaなどを再インストールし直したりしているものの、同じエラーが出てきます。

考えられる原因やどこか手順に間違いがあるなどがあれば、教えていただきたいです><

os windows10
python3.5.2
gpu gtx 1050ti

Answer 1

Installing TensorFlow on Windows

公式の説明によると、Windows版のTensorflowは、現状 cuDNN 5.1 でないと動きません。

Comments

[MasashiKimura]

ついでにいうと、GTX 1050Ti はメモリが 4GBしかないので、本格的なネットワークは OOM でうごかないでしょうね。

[退会済みユーザー]

ありがとうございます！
cuDNN5.1を入れたら無事動きました
予算の都合上1050tiを選びました
メモリが足りなかったら特定の箇所のみcpuに演算させる事も出来るようなのでとりあえずはそれでしのごうかと思います