【keras-rl,colab】keras-rlをcolabのgpuを用いて実行する

keras-rlをcolab上で動かせるようにしたのでメモしておきます。

Colaboratory(colab)

Colaboratory(colab)はgoogleが提供してくれるJupyterノートブック環境です。
無料でGPUが使えるので、非常にありがたいサービスです。
最近ではvimでも操作ができるようになったので本当に感謝です。

Google Colaboratory

keras-rlとは

Keras-rlとは 深層学習用ライブラリであるkerasを用いて、深層強化学習のアルゴリズムを実装したライブラリです。

keras-rl/keras-rl
Deep Reinforcement Learning for Keras. Contribute to keras-rl/keras-rl development by creating an account on GitHub.

容易に深層強化学習を試すことができます。

とりあえずローカルでkeras-rlを試したい場合は以下の記事を参照していただければと思います。

【Deep Q Learning、深層強化学習】keras-rlを試してみる
今回は深層強化学習ライブラリのKeras-rl をとりあえず試したいと思います。基本的なpythonの操作はできることを想定しています。keras-rlとはKeras-rlとは 深層学習用ライブラリであるkerasを用いて、深層強化...

colabのgpuを用いてkeras-rlを動かしてみる(atari)のブロック崩し)

以下のatariのブロック崩しをkeras-rlのdqnで解いてみました。

f:id:ttt242242:20190824131849j:plain

出典: https://gym.openai.com/envs/Breakout-v0/

dqnに関しては以下の記事を参照してください。

【深層強化学習,初心者向け】Deep Q Networkとは 〜図を使って説明〜
**Q Network**

GPUの有効化

実行する前にcolabでGPUを使えるようにcolabの設定を変更する必要があります。
手順は以下のようになります。

  1. 「ランタイム」
  2. 「ランタイムのタイプの変更」
  3. ハードウェアアクセラレータでGPUに変更

ソースコード

以下にソースコードを示します。

参考文献

つくりながら学ぶ! 深層強化学習 ~PyTorchによる実践プログラミング~

コメント

タイトルとURLをコピーしました