edo1z blog

プログラミングなどに関するブログです

Windows10にDockerを使ってTensorFlowをインストールする

Windows10にTensorFlowをインストールします。Dockerを使うと簡単にインストールできます。GPUなしバージョンをインストールします。Dockerはインストールされているものとします。ちなみに、Windows環境だとTensorFlowのGPU版を使うことは難しいようです。今度、最近できるようになった、Windows10上で、Linuxが動かせるやつを試してみようと思います。 参考:Windows10上のTensorFlowでGPUを利用しようとしてダメだった

TensorFlowが入ってるDockerコンテナを実行

これだけでTensorFlowが使えるようになります。

$ docker run -it b.gcr.io/tensorflow/tensorflow:latest-devel

動くかチェック

下記コードを黒い画面でやってみてそのとおりになったら、動いてます。 参考:Run TensorFlow from the Command Line

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>