Keras（五）——指定显卡且限制显存用量（GPU/CPU使用）

Keras在使用GPU时有个特点，就是默认全部占满显存。

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 使用cpu
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1" #使用多块GPU

参考：

https://cloud.tencent.com/developer/article/1434932