Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';

解決了題目所述問題。 報錯完整信息爲: 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_
相關文章
相關標籤/搜索