RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)
训练模型的GPU和你指定加载模型地GPU不同,或是多gpu并行模型加载时,容易报此错误
Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';
最新推荐文章于 2024-02-27 14:47:33 发布