Ubuntu 18.04 LTS で NVIDIA-Docker2 を使ってみる

Docker Unix Linux 機械学習 Ubuntu18.04LTS TensorFlow DeepLearning

ニューラルネットワークに代表される機械学習の一部のアルゴリズムでは、学習する上で大量の行列演算を必要とする。そこで、行列演算を高速化する目的で GPGPU を利用することが一般的になっている。そして、この分野では機械学習のライブラリが GPGPU の …

2017-06-07

Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する

DeepLearning Keras TensorFlow Ubuntu16.04LTS 機械学習

Keras のバックエンドに TensorFlow を使う場合、デフォルトでは一つのプロセスが GPU のメモリを全て使ってしまう。今回は、その挙動を変更して使う分だけ確保させるように改めるやり方を書く。環境には次のようにしてセットアップした Ubuntu 16.04 LTS …

2017-03-13

Python: Keras/TensorFlow の学習を GPU で高速化する (Ubuntu 16.04 LTS)

DeepLearning Ubuntu16.04LTS 機械学習 TensorFlow Keras Python

以前、このブログで Keras/TensorFlow の学習スピードを GPU を使って速くする記事を書いた。ただし、このとき使った OS は Mac OS X (macOS Sierra) だった。 blog.amedama.jp とはいえ NVIDIA の dGPU を積んだ Mac がどれだけあるんだというと、正直なか…