Colab降级Python3.6后安装Torch1.2.0提示模块不存在的问题

核心原因分析 #

出现No module named 'torch'的问题，主要集中在两个关键点：

1. pip与Python版本不匹配：你用pip install安装的Torch可能被放到了系统原Python版本的依赖目录里，而非切换后的Python3.6环境。Colab默认的pip大概率仍绑定旧版Python，没同步到你刚配置的Python3.6。
2. Colab运行时内核未同步切换：Colab的Jupyter内核独立于系统Python配置，即便你改了系统默认Python版本，当前执行代码的内核可能还是用Colab原始的Python版本（比如3.8+），自然找不到装在Python3.6里的Torch。

针对性解决步骤 #

1. 用Python3.6专属pip重新安装Torch #

卸载残留的Torch后，明确指定Python3.6的pip来安装，避免版本混淆：

# 卸载残留Torch
!python3.6 -m pip uninstall torch -y
# 重新安装Torch 1.2.0
!python3.6 -m pip install torch-1.2.0+cpu-cp36-cp36m-manylinux1_x86_64.whl

2. 验证Python与pip的绑定关系 #

执行以下命令确认pip是否对应Python3.6：

# 查看默认pip对应的Python版本
!pip --version
# 查看Python3.6专属pip的版本
!python3.6 -m pip --version

如果两者输出的Python版本不一致，后续所有包安装操作都要用python3.6 -m pip替代pip，确保依赖装到正确环境。

3. 切换Colab运行时内核（关键） #

Colab的Jupyter内核不会自动跟随系统Python变化，需手动配置：

• 点击顶部菜单栏「Runtime」→「Change runtime type」
• 查看「Python interpreter」选项，若没有Python3.6，执行以下代码安装并添加专属内核：

!python3.6 -m pip install ipykernel
!python3.6 -m ipykernel install --user --name=python36

安装完成后刷新页面，回到「Change runtime type」选择python36内核，再执行import torch即可。

4. 验证Torch安装路径 #

执行以下命令确认Torch是否安装在Python3.6的依赖目录：

!python3.6 -c "import torch; print(torch.__file__)"

如果能输出具体路径，说明安装成功，切换到对应内核就能正常导入。

内容的提问来源于stack exchange，提问作者Luiz Fernando Puttow Southier