Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when

在学习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),改成

即可。

 

你可能感兴趣的:(pytorch,深度学习,人工智能)