解决RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0])
解决RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1问题在加载多GPU模型后使用pytorch的DataParallel()时出现以上报错,网上查找了很多资料都没有正确的解决方法。个人...
原创
2019-08-02 12:06:13 ·
11521 阅读 ·
9 评论