2022年11月19日
pytorch 使用单个GPU与多个GPU进行训练与测试
摘要： pytorch 使用单个GPU与多个GPU进行训练与测试 使用多个GPU的原理就是通过上面这句代码将model在每个GPU上分别保存一份，然后对model的输入tensor进行自动的分割，每个GPU计算tensor的一部分，这样就能实现计算量的平均分配。在每个model计算完成之后，DataPara 阅读全文
posted @ 2022-11-19 16:36 watsono 阅读(206) 评论(0) 推荐(0) 编辑