トップ «前の日記(2017/06/05) 最新 次の日記(2017/06/08)» 編集

kaztomo日記


2017/06/06

_ [Python]Tensorflow環境を整える

すでに導入している 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

最終的に行った手順は以下の通り

  1. NVIDIA CUDA 8.0 Toolkitをインストールする。(なんと 1.3GB!!)
  2. 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 へまるごと展開
  3. tensorflow パッケージを導入する
C:> pip install --upgrade tensorflow
C:> pip install --upgrade tensorflow-gpu
  1. 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
  1. 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!'

参考にさせて頂いたサイト: