Jupyter Lab出现ModuleNotFoundError: No module named 'torch'问题求助

核心原因 #

Jupyter Lab默认使用的Python内核，和你安装PyTorch的环境（全局环境/虚拟环境）不匹配，导致找不到已安装的torch库。

具体解决步骤 #

1. 定位安装了torch的Python环境路径 #

在能正常导入torch的Python终端中执行以下代码：

import sys
print(sys.executable)

会得到类似/usr/bin/python3或/home/xxx/miniconda3/envs/myenv/bin/python的路径，这就是目标环境的位置。

2. 给Jupyter Lab添加对应内核 #

• 先激活安装了torch的环境（以conda环境为例）：

  conda activate your_env_name
  

  如果你用的是venv虚拟环境，执行：

  source your_env_name/bin/activate
  

• 在该环境下安装ipykernel：

  pip install ipykernel
  

• 将该环境注册为Jupyter可识别的内核：

  python -m ipykernel install --user --name=your_env_name --display-name="Python (your_env_name)"
  

3. 在Jupyter Lab中切换内核 #

打开Jupyter Lab后，点击顶部菜单栏的Kernel -> Change Kernel，选择刚才注册的Python (your_env_name)内核，重新运行导入torch的代码即可。

备选方案：直接在Jupyter Lab终端安装torch #

如果不想切换内核，也可以打开Jupyter Lab的终端，执行和你之前一致的torch安装命令，比如：

pip install torch torchvision torchaudio

内容的提问来源于stack exchange，提问作者Gianni Spear