安装torch 1.4.0后无法导入torch.optim等子模块的问题如何解决
PyTorch 1.4.0 部分子模块导入失败解决方案
- 验证Jupyter内核匹配性
在当前报错的Jupyter Notebook单元格中运行如下代码,确认输出的Python执行路径与安装PyTorch 1.4.0的虚拟环境路径完全一致:
若路径不匹配,点击顶部菜单栏「Kernel」→「Change Kernel」,选择对应虚拟环境的内核后重试即可。import sys print(sys.executable) - 排查本地命名冲突
检查当前运行Notebook的工作目录下,是否存在命名为torch.py的文件,或者名为torch的文件夹。Python导包时会优先加载当前目录下的同名文件/文件夹,覆盖官方PyTorch库,将该类文件/文件夹重命名或移出工作目录即可。 - 验证PyTorch包完整性
激活对应虚拟环境后,在系统终端中依次执行如下操作:- 运行
pip show torch,检查返回的版本是否为1.4.0,安装路径是否属于当前虚拟环境 - 在终端启动Python交互环境,执行
import torch.optim,若终端也出现导入报错,说明PyTorch包已损坏,执行如下命令重装对应版本即可:
pip install torch==1.4.0 torchvision==0.5.0 - 运行
- 修复Jupyter内核配置
若终端中可正常导入所有子模块,仅Jupyter中报错,说明内核配置异常,激活对应虚拟环境后执行如下命令重新注册内核:
执行完成后重启Jupyter Notebook,选择新注册的内核运行代码即可。pip install ipykernel python -m ipykernel install --user --name=替换为你的虚拟环境名称
内容的提问来源于stack exchange,提问作者Anvar Ganiev
相关产品推荐
相关产品推荐

