Pythonのインストール
TensorFlowは最新の3.6に対応していないので3.5以前のバージョンをダウンロードする。(2017/09/25現在)
Pythonダウンロードページ
TensorFlow CPU版とCPU版
CPU版とGPU版の2種類があり、Nvidia製グラフィックボードを搭載している場合はGPU版が使用できる。
(2017/09/25現在、AMD製GPUは対応していない模様)
CPUでは、学習処理にかなり時間がかかるためGPU版がおすすめ。
前準備
必要なライブラリをインストール
※pipからの通常インストールではエラーが出るため下記を使用
wheel
pip install wheel
numpy+mkl
pip install numpy‑1.13.1+mkl‑cp35‑cp35m‑win_amd64.whl
scipy
pip install scipy‑0.19.1‑cp35‑cp35m‑win_amd64.whl
GPU版を使用する場合
-
CUDAをインストール
ver.8をインストール -
cuDNNをインストール
ダウンロードにはNVIDIA Developer Programへの登録が必須
TensorFlowは最新版のcuDNNに対応していないようなので5.1をインストール
TensorFlowをインストール
CPU版
pip install tensorflow
GPU版
pip install tensorflow-gpu
Kerasをインストール
pip install keras
構築した環境
Windows 10 64bit
Nvidia Geforce GTX 1050Ti
Python(x64) 3.5.4
wheel 0.30.0
numpy 1.13.1+mkl
scipy 0.19.1
TensorFlow-gpu 1.3.0
Keras 2.0.8