从GitHub下载了一个项目代码,配置好环境后运行(由于我只有集显,GPU能力不够,我用的pytorch是CPU版的)运行不成功,结果报错。
【请使用torch。加载map_location=torch.device('cpu')来将你的存储映射到cpu】
这句话是什么意思?怎么加载map_location=torch.device('cpu')?
本人小白一个,望不吝赐教。
报错信息:
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。
RuntimeError:试图在CUDA设备上反序列化对象,但torch.cuda.is_available()为False。如果您运行的是一台只使用cpu的机器,请使用torch。加载map_location=torch.device('cpu')来将你的存储映射到cpu