2020年7月29日
torch.nn.DataParallel使用细节
摘要： 仅使用nn.DataParallel，gpu0和gpu1、gpu0和gpu2、gpu0和gpu3等包含gpu0的组合都是可以的，其余组合不行，报错RuntimeError: module must have its parameters and buffers on device cuda:1 (d 阅读全文
posted @ 2020-07-29 11:05 zonechen 阅读(6386) 评论(0) 推荐(0) 编辑