分类：显存占用

2020
10-08

Keras - GPU ID 和显存占用设定步骤

初步尝试Keras(基于Tensorflow后端)深度框架时,发现其对于GPU的使用比较神奇,默认竟然是全部占满显存,1080Ti跑个小分类问题,就一下子满了.而且是服务器上的两张1080Ti.服务器上的多张GPU都占满,有点浪费性能.因此,需要类似于Caffe等框架的可以设定GPUID和显存自动按需分配.实际中发现,Keras还可以限制GPU显存占用量.这里涉及到的内容有：GPUID设定GPU显存占用按需分配GPU显存占用限制GPU显存优化1.... 继续阅读 >