PyTorch数据并行nn.DataParallel
仅使用nn.DataParallel,gpu0和gpu1、gpu0和gpu2、gpu0和gpu3等包含gpu0的组合都是可以的,其余组合不行,报错RuntimeError: module must have its parameters and buffers on device cuda:1 (device_ids[0]) but found one of them on device: cud...
原创
2019-08-29 14:38:24 ·
18525 阅读 ·
0 评论