RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/86405099

RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'

原因:变量没有加cuda

data=data.cuda()

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/86405099
今日推荐