运行psmnet时候报错不知道怎么解决

我的电脑是核显，pytorch安装的就是CPU版本的 该怎么该psm 的代码才能运行呢我把model.cuda()改成model.cpu()也不行还需要改什么么

默认改成True，其他地方先不改，不就是没有cuda训练吗？

或者这里的改成args.cuda=False
然后就是下面的带.cuda()的全部去掉，Dataparallel这句注释掉，改成cpu加载