tensorflow使用GPU训练时的显存占用问题

http://blog.csdn.net/sunwuhuang1/article/details/53946462 这篇文章很有用。 最近用GoogleNet (inception-v3)做fine-tune训练时,遇到了一个问题。当我选择单个GPU训练时候,服务器中所有的GPU显存都会被占满,如下图所示:     出现以上问题的原因是tensorflow在训练时默认占用所有GPU的显存。    
相关文章
相关标签/搜索