安装PyTorch 0.4.0(cuda9.0)后import torch报ModuleNotFoundError如何解决

排查步骤 #

• 验证conda环境一致性
  执行conda info --envs查看所有conda环境，核对当前激活的环境，确认pytorch的安装环境与运行python代码的环境为同一个。
• 检查pytorch安装状态
  激活对应环境后执行conda list | grep pytorch，查看输出中是否存在版本为0.4.0、适配cuda90的pytorch安装条目，无条目则说明实际未安装成功。
• 核对python解释器路径
  执行which python（Linux/macOS）或where python（Windows），确认当前调用的python解释器路径归属已激活的conda环境，避免调用系统默认python或其他环境的解释器。
• 检查安装源可用性
  低版本PyTorch依赖旧镜像资源，执行conda config --show channels确认安装源可正常访问适配cuda9.0的旧版本资源。

解决方案 #

• 环境不匹配的情况
  执行conda activate <安装pytorch的目标环境名>切换到对应环境后，再运行python测试导入。
• 实际未安装成功的情况
  1. 先执行conda uninstall pytorch cuda90清理残余安装文件
  2. 优先选择pip方式安装适配cuda9.0的0.4.0版本，执行对应官方旧版whl安装命令即可
  3. 安装完成后重新执行conda list | grep pytorch核对版本信息
• 解释器路径不匹配的情况
  切换到conda环境对应的解释器执行代码，若使用IDE则手动将项目解释器配置为对应conda环境下的python解释器。
• 安装源无法访问的情况
  安装时临时指定国内镜像源路径即可完成正常安装。

内容的提问来源于stack exchange，提问作者Aman Savaria