显存控制
文章平均质量分 74
silent56_th
大学生一枚,对机器学习、图像处理感兴趣,偶尔也会刷刷leetcode或者hihocoder。希望在这里可以和大家多多交流,互相学习。
展开
-
使用GPU训练大神经网络之 显存控制策略与提升
本文是“[PPoPP18] SuperNeurons:Dynamic GPU Memory Management for Training Deep Neural Networks”的论文阅读笔记,论文、talk视频PPT等详见作者主页:https://linnanwang.github.io/ 参考资料: https://linnanwang.github.io/ How to Optim...原创 2018-05-21 21:26:37 · 7389 阅读 · 3 评论 -
python获取GPU相关信息
笔者想要根据空闲显存信息,动态分配tensorflow中session运行的GPU。搜索下来没有发现成熟的方案,备选方案记录如下: 方案一: 翻译自:How to get current available GPUs in tensorflow? 使用tensorflow的后端函数device_lib.list_local_devices()。 缺点1:会尝试并占用所有可用GPU显存,需要...原创 2018-07-31 20:28:17 · 7817 阅读 · 0 评论