报错:RuntimeError: すべてのテンソルが同じデバイス上にあることが予期されましたが、少なくとも 2 つのデバイスが見つかりました、cuda

詳細なエラー メッセージは次のとおりです。

RuntimeError: Expected all tensors to be on the same device, 
but found at least two devices, cuda:0 and cpu! 
(when checking argument for argument mat1 in method wrapper_mm)

エラー報告場所

エラーが報告される場所: ネットワークの転送関数 (linear.py/forward)
ここに画像の説明を挿入します

解決

ここに画像の説明を挿入します

おすすめ

転載: blog.csdn.net/gls_nuaa/article/details/132823095