多个GPU训练的问题：

[mufanxing]

谢谢大佬的分享， 这有个问题想请教您一下：我想多个GPU训练这个模型，设置了多个GPU训练，但是训练时显示“RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)”，好像是模型定义不规范的问题，导致权重与模型在不同GPU上，请大佬指点一二，谢谢啦

[hellloxiaotian]

你好： 很抱歉晚回复你了！可以通过CUDA指定GPU方式来设置多卡使用，这样的操作是简单和有效的。 再次感谢你对我们工作的关注！ 祝好！ 春伟 副教授 西北工业大学 在 2021-06-24 16:06:58，"ziyouzizai111" ***@***.***> 写道： 谢谢大佬的分享， 这有个问题想请教您一下：我想多个GPU训练这个模型，设置了多个GPU训练，但是训练时显示“RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)”，好像是模型定义不规范的问题，导致权重与模型在不同GPU上，请大佬指点一二，谢谢啦 — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.