将从GPU获取的pickle文件加载到CPU
将从GPU获取的pickle文件加载到CPU
提问于 2019-07-18 01:56:09
当我加载一个pickle文件到CPU时，我遇到了一个问题。我在网上搜索了一下，他们说我需要添加map_location参数。但是，在我添加了这个参数之后，问题仍然存在。
代码如下：
torch.__version__
torch.load('featurs.pkl',map_location='cpu')
>>>
'1.0.1.post2'
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='cpu'
to map your storages to the CPU.我知道这是因为不同的设备，但我使用了错误消息中的指令，所以我不知道下一步如何解决它。
提前感谢！
回答 1
Stack Overflow用户
发布于 2019-07-18 02:16:13
试试这个：
torch.load('featurs.pkl',map_location=torch.device('cpu'))页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/57081727
复制相关文章
相似问题
腾讯云开发者
