已安装PyTorch但Jupyter Notebook提示No module named 'torch'的解决方法

• 核对kernel与虚拟环境的Python路径

  1. 在命令行激活scan_venv虚拟环境，执行：
     • Linux/macOS：which python
     • Windows：where python
       记下输出的Python路径。
  2. 在Jupyter Notebook的scan_env kernel中运行代码：

     import sys
     print(sys.executable)
     

     对比两个路径，若不一致，说明kernel关联的环境有误，需重新创建kernel。

• 重新关联虚拟环境到Jupyter kernel

  1. 命令行激活scan_venv：
     • Conda环境：conda activate scan_venv
     • 虚拟env（Linux/macOS）：source scan_venv/bin/activate
     • 虚拟env（Windows）：scan_venv\Scripts\activate
  2. 安装/更新ipykernel：pip install --upgrade ipykernel
  3. 删除旧的scan_env kernel：jupyter kernelspec remove scan_env
  4. 重新创建关联kernel：python -m ipykernel install --user --name scan_env --display-name "scan_env"

• 确认PyTorch安装在当前虚拟环境

  1. 激活scan_venv后执行：
     • Linux/macOS：pip list | grep torch
     • Windows：pip list | findstr torch
       若没有torch条目，说明之前的安装不在这个环境里，需重新安装：

     # 根据你的系统和CUDA版本调整，示例为CUDA 11.8
     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
     

• 重启Jupyter Notebook
  关闭所有Jupyter进程，重新启动后选择scan_env kernel，再次尝试import torch。

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