PyTorch1.7.1无法导入_worker_manager_loop/_pin_memory_loop求解决方案
PyTorch 1.7版本对应函数导入方案
- PyTorch 1.1及后续版本重构了
torch.utils.data.dataloader的内部结构,_pin_memory_loop已经从原路径移除,迁移到了内部工具包路径下。 - 1.7.1版本的正确导入语句为:
from torch.utils.data._utils.pin_memory import _pin_memory_loop
- 替换导入语句后,原代码中使用该函数的逻辑无需修改,两个版本的函数核心功能、调用参数保持兼容。
- 若仍出现导入异常,可执行以下代码查看当前环境下对应模块的成员列表,确认函数命名:
import torch.utils.data._utils.pin_memory print(dir(torch.utils.data._utils.pin_memory))
注意:以下划线开头的函数属于PyTorch内部私有API,官方不保证跨版本兼容性,生产环境建议优先使用公开稳定API实现对应功能,避免后续版本升级出现兼容问题。
内容的提问来源于stack exchange,提问作者Nour
相关产品推荐
相关产品推荐

