通过设置Keras的Tensorflow后端的全局变量达到。使用过程中显示的设置session:. 我就废话不多说了,大家还是直接看代码吧!

Keras设定GPU使用内存大小方式(Tensorflow backend)