按照网上的方法,先对比出错
https://blog.csdn.net/qq_28935065/article/details/119007334
其次还是不行的话,将
import torch
from torch.utils,data import DataLoader
等torch语句放到device = torch.device("cuda"if torch.cuda.is_available() else "cpu")后面
接着,输入input的内容和权重参数有的在cpu,有的在gpu上
https://blog.csdn.net/qq_40520596/article/details/106980556
或者