pytorch 多gpu 1号卡占用问题

发表于 2018-03-Thu | 阅读次数:

在使用pytorch multiple gpu的时候，1号卡的实际占用会大于分配给它的。

所以需要修改源代码使得1号卡分配的时候减少一个batch.

具体修改见链接

http://jeff-leaf.site/2018/01/03/Torch%E5%A4%9AGPU/#more

请作者喝一杯咖啡☕️