如何设置Keras使用特定GPU ID及优化显存占用？

本文共计1088个文字，预计阅读时间需要5分钟。

初试Keras（基于Tensorflow后端）深度学习框架时，发现其对GPU的使用简直神奇。默认情况下，模型会全部占用显存，1080Ti显卡也能轻松跑出小分类问题。服务器上更是有两张1080Ti，轻松应对。