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

原因：变量没有加cuda

 

data=data.cuda()

posted on 2019-09-03 17:19  fuge92  阅读(6497)  评论(0)  编辑  收藏  举报