バックエンドをTensorFlowとしてKerasを利用しようとすると,デフォルトだとGPUのメモリを全部使う設定になっていて複数の実験を走らせられないので,GPUメモリの使用量を抑える設定方法について紹介します.1 2 検証環境 GPUのメモリ使用量を抑える 下記コードを貼り付けるなり,importするなりすれば大丈夫です. 最小限のGPUメモリのみ確保 gpu_options.allow_growthで設定可能です. 実行時に必要な分だけ確保する方法で,さらに必要になるとメモリ領域を拡張するようになっています. ただし自動的にはメモリを開放しないので,メモリが断片化して性能が悪化する可能性があり注意が必要です.1 import tensorflow as tf from keras import backend as K config = tf.ConfigProto() config
![KerasでGPUメモリの使用量を抑える方法](https://cdn-ak-scissors.b.st-hatena.com/image/square/1a92f639245c4ffaedb0607b51f357d8dc840ede/height=288;version=1;width=512/https%3A%2F%2Fqiita-user-contents.imgix.net%2Fhttps%253A%252F%252Fcdn.qiita.com%252Fassets%252Fpublic%252Farticle-ogp-background-412672c5f0600ab9a64263b751f1bc81.png%3Fixlib%3Drb-4.0.0%26w%3D1200%26mark64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZ3PTk3MiZoPTM3OCZ0eHQ9S2VyYXMlRTMlODElQTdHUFUlRTMlODMlQTElRTMlODMlQTIlRTMlODMlQUElRTMlODElQUUlRTQlQkQlQkYlRTclOTQlQTglRTklODclOEYlRTMlODIlOTIlRTYlOEElOTElRTMlODElODglRTMlODIlOEIlRTYlOTYlQjklRTYlQjMlOTUmdHh0LWFsaWduPWxlZnQlMkN0b3AmdHh0LWNvbG9yPSUyMzIxMjEyMSZ0eHQtZm9udD1IaXJhZ2lubyUyMFNhbnMlMjBXNiZ0eHQtc2l6ZT01NiZzPWIxZmFlZWJlYThhZTdiNWQ2MmRjZDg0MzFhYmE1Y2Vi%26mark-x%3D142%26mark-y%3D57%26blend64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZoPTc2Jnc9NzcwJnR4dD0lNDBuYW1ha2Vtb25vJnR4dC1jb2xvcj0lMjMyMTIxMjEmdHh0LWZvbnQ9SGlyYWdpbm8lMjBTYW5zJTIwVzYmdHh0LXNpemU9MzYmdHh0LWFsaWduPWxlZnQlMkN0b3Amcz01ZjFkMDlhZDg1NmM3ZjczZjMyMTQzOGM5MDEwNTEyYw%26blend-x%3D142%26blend-y%3D486%26blend-mode%3Dnormal%26s%3Dbdd4d7c656988ffe4628fe1f1c5bf2f5)