【问题解决】keras指定GPU

一、问题描述

在keras训练模型时，制定了使用三块GPU但是实际运行的时候却把所有GPU全部占满了。

原始代码如下

二、解决

查找原因，发现Keras使用显卡时是默认调用所有的GPU，并且占满所有显存的。所以最好将指定GPU的代码写在improt keres 和 import tensorflow之前，不然可能出错。

修改后将

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2"

写在了代码的最开始，问题解决！