2017/06/06
すでに導入している Anaconda が 4.3.0のために Tensorflow で使えない Python3.6環境だったり、 Quadra を使いこなすために Tensorflow-gpu を導入しようとして失敗していたり、
ImportError: No module named '_pywrap_tensorflow_internal'
なんてエラーが出ていたり、他にも問題があったりしたけど、ようやっと環境を整えた。
Jupyter Notebook からも使えている感じ。\(^o^)/
結局のところ、公式ドキュメント をちゃんと読もうという結果だった (^^;;
私の仕事場での Python環境:
- Windows7 pro 64bit / Core i7-6700K
- ELSA Quadro2000M
- CUDA v8.0
- cuDNN v5.1
最終的に行った手順は以下の通り
- NVIDIA CUDA 8.0 Toolkitをインストールする。(なんと 1.3GB!!)
- cuDNN Downloadサイトから、cuDNN v5.1 をダウンロードする。
- NVIDIA Developer Membership に登録しないとダウンロードできないので、未登録な場合は新規登録する。
- リストの先頭にある最新版 v6.0 を間違ってダウンロードしないこと。
- →こっち。Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0
- ダウンロードした圧縮ファイルを、C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 へまるごと展開
- tensorflow パッケージを導入する
C:> pip install --upgrade tensorflow C:> pip install --upgrade tensorflow-gpu
- Anaconda3 にて tensorflow 用の仮想環境を作成する
C:> conda info -e # conda environments: # root * C:\Anaconda3 C:> conda create -n tensorflow C:> activate tensorflow (tensorflow)C:> (tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.1.0-cp35-cp35m-win_amd64.whl (tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl
- Pythonを起動してインストールできているか確認する
(tensorflow) C:>python
Python 3.5.3 |Continuum Analytics, Inc. (snip)
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
2017-06-06 HH:MM:SS W c:\tf_jenkins ... library wasn't compiled to use SSE instructions,..
2017-06-06 HH:MM:SS W c:\tf_jenkins ... library wasn't compiled to use SSE2 instructions,...
(snip)
2017-06-06 HH:MM:SS W c:\tf_jenkins ... library wasn't compiled to use FMA instructions,...
2017-06-06 HH:MM:SS I c:\tf_jenkins ... Found device 0 with properties: name: Quadro M2000 ...
pciBusID 0000:01:00.0
Total memory: 4.00GiB
Free memory: 3.87GiB
2017-06-06 HH:MM:SS I c:\tf_jenkins ... DMA: 0
2017-06-06 HH:MM:SS I c:\tf_jenkins ... 0: Y
2017-06-06 HH:MM:SS I c:\tf_jenkins ... Creating TensorFlow device (/gpu:0) -> (device: 0, name: Quadro M2000, pci bus id: 0000:01:00.0)
>>>
>>> print(sess.run(hello))
b'Hello, TensorFlow!'
参考にさせて頂いたサイト:
[ツッコミを入れる]