在学习pytorch的时候,在进行GPU的训练的时候出现这个问题,特此记录一下这个问题的解决,
首先应该找到报错的位置,比如的报错位置和语句是
其中点击蓝色到报错位置值,图片的第二行是报错的内容,我在查询别的资料的时候发现
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_addmm)这个问题是没有把运行内容放到GPU上,
解决办法:
先看自己写这个语句没有
之后看看自己的问题,比如我的问题是
我需要把内容搬到GPU上运行,在内容后加上.to(device),改成
即可。