Ubuntu 18.04下Jupyter提示torch模块找不到（CLI已安装）的解决方法

核对Jupyter的Python环境路径 #

• 在Jupyter Notebook里运行以下代码，查看当前使用的Python路径：

  import sys
  print(sys.executable)
  

• 在Ubuntu命令行执行which python3.6，对比两个路径是否一致。若不一致，说明Jupyter使用的是未安装torch的其他Python环境。
• 修复步骤：
  1. 激活已安装torch的Python环境（虚拟环境需先激活）
  2. 安装ipykernel：pip install ipykernel
  3. 将该环境添加到Jupyter内核列表：python -m ipykernel install --user --name=自定义环境名
  4. 重启Jupyter，在Kernel菜单中切换到刚添加的环境内核

直接在Jupyter对应环境安装torch #

• 打开Jupyter终端（New -> Terminal），执行适配Python3.6、CUDA10.2的torch安装命令：

  pip install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio==0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
  

确认pip与Python版本匹配 #

• 在命令行执行pip --version，查看其关联的Python版本是否为3.6。若存在多个pip版本（如pip3），安装torch时需使用对应版本：pip3 install torch==1.10.2+cu102 ...
• Jupyter终端中也可直接用pip3执行安装命令

在目标环境重新安装Jupyter #

• 激活已安装torch的Python环境，执行pip install jupyter
• 在该环境下启动Jupyter（命令行执行jupyter notebook），此时Jupyter会默认使用当前环境的Python解释器

内容的提问来源于stack exchange，提问作者Lee na-hyeon