阅读背景:

keras指定运行时显卡及限制GPU用量

来源:互联网 

keras在使用GPU的时候有个特点,就是默认全部占满显存。 这样如果有多个模型都需要使用GPU跑的话,那么限制是很大的,而且对于GPU也是一种浪费。因此在使用keras时需要有意识的设置运行时使用那块显卡,需要使用多少容量。keras在使用GPU的时候有个特点,就是默认全部占满显存。 这样如果有多个模型都需要




你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: