Keras と GPU版 TensorFlow をインストール
(tensorflow-with-gpu) $ pip install keras tensorflow-gpu
(tensorflow-with-gpu) $ pip list --format=columns | grep -i -e keras -e tensorflow
Keras 2.0.8
tensorflow-gpu 1.3.0
tensorflow-tensorboard 0.1.5
sudo apt install nvidia-cuda-dev
sudo apt update
sudo apt upgrade
33.
git clone keras
>cd
> mkdir src
> cd src
> git clone https://github.com/fchollet/keras.git
> cd keras/examples
34.
mnist サンプルを実行
## 時間計測を行いmnist cnn を実行する
> time python mnist_cnn.py
Test loss: 0.0300338261104
Test accuracy: 0.9902
real 2m5.097s
user 0m54.152s
sys 1m15.292s
35.
GPU稼働情報を表示
> nvidia-smi
Sat Sep2 18:33:33 2017
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.69 Driver Version: 384.69 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 105... Off | 00000000:01:00.0 Off | N/A |
| N/A 43C P0 N/A / N/A | 398MiB / 4041MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1129 G /usr/lib/xorg/Xorg 224MiB |
| 0 1848 G compiz 135MiB |
| 0 2712 G ...el-token=A824897B35D121CF49ACBD4502480B90 36MiB |
| 0 3840 G /usr/lib/firefox/firefox 1MiB |
+-----------------------------------------------------------------------------+
36.
今度はCPUで計測
$ pip uninstall-y tensorflow-gpu
$ pip install tensorflow
$ time python mnist_cnn.py
Test loss: 0.0295431383794
Test accuracy: 0.9893
real 18m30.578s
user 122m28.916s
sys 10m3.364s