分类：资源耗尽

2020
10-08

解决Keras使用GPU资源耗尽的问题

我们在使用GPU资源进行训练的时候，可能会发生资源耗尽的情况，那么在在这种情况，我们需要对GPU的资源进行合理的安排，具体使用办法如下：框架：Tensorflow和Keras方法importtensorflowastfimportkeras.backend.tensorflow_backendasKTFconfig=tf.ConfigProto()config.gpu_options.allow_growth=True#不全部占满显存,按需分配sess=tf.Session(config=config)KTF.set_session(sess)#设置session通过这种方法，就能合... 继续阅读 >