TensorFlowをIntellij IDEAで使う

Intellij IDEAでTensorFlowを扱うまでの設定について記載。

Python2.7が入っている前提です。
pip virtualenvのインストール

$ sudo easy_install pip
$ sudo pip install --upgrade virtualenv


virutalenvで環境を作成

$ virtualenv --system-site-packages ~/tensorflow
$ cd ~/tensorflow


activate

$ source bin/activate # If using bash
$ source bin/activate.csh # If using csh


activateすると表記が以下にかわります

(tensorflow)$


今回はMAC CPU版のみでインストールします

(tensorflow)$ pip install --upgrade https://storage.googleapis.com/tensorflow/mac/tensorflow-0.5.0-py2-none-any.whl


pythonを立ち上げて

$ python


以下のコードが実行できれば取り急ぎ、tensorflowが動いています。

import tensorflow as tf
hello = tf.constant('HelloWorld')
sess = tf.Session()
print sess.run(hello)


ここからIntellij IDEAの設定

File>New Project>Python Moduleを選択

Configure SDKの画面で「+」ボタンを押す

「Pyhton SDK」を選択する

「select interpreter path」の中から、上記で作成した~/tensorflow...を選択する。

SDKとして追加されるのでOKしてfinishまで進む

pythonファイルを作成して先ほどのimport tensorflow〜
のコードを記述する

Run>Runから先ほど作成したファイルを選択

HelloWorldがコンソールに出ればOK

以上