2017年11月14日
tensorflow 如何限制显存大小
摘要： Python在用GPU跑模型的时候最好开多进程，因为很明显这种任务就是计算密集型的。 用进程池好管理，但是tensorflow默认情况会最大占用显存，尽管该任务并不需要这么多，因此我们可以设置显存的按需获取，这样程序就不会死掉了。 1. 按比例预留： tf_config = tensorflow.C 阅读全文
posted @ 2017-11-14 19:22 安阳小栈-客官歇会吧 阅读(4446) 评论(0) 推荐(0) 编辑