RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device(‘cpu’) to map your storages to the CPU.解决办法
首先找到出错语句
我这里是
net.load_state_dict(torch.load(args.trained_model))
改为
net.load_state_dict(torch.load(args.trained_model,map_location='cpu'))
即可
主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我
点个赞支持一下吧