強化学習を行っていますが、パフォーマンスに問題があります。
状況、カスタム コードなし:
- Google Cloud にGoogle Deep Learning VM ( https://console.cloud.google.com/marketplace/details/click-to-deploy-images/deeplearning ) をロードしました。これには、NVidia K80 ビデオカードとともにすべての前提条件 (CUDA、cuDNN、ドライバー) がインストールされています。
- keras-rl、OpenAI ジムをインストール
- ここで、(標準) 例のdqn_cartpole.pyを 46 行目で Visualize=False を指定して実行すると、GPU の約 20% が使用され、1 秒あたり約 100 ステップになり、Razer Blade で CPU を使用するよりも約 3 倍遅くなります。 15 (i7-8750H)。
- 考えられるすべてのボトルネックを確認しましたが、CPU 使用率、メモリ、HD I/O はすべて正常です。
助けてください！
前もって感謝します