SSH连接校内虚拟机中Jupyter安装库后无法导入的问题

问题本质 #

你遇到的是Jupyter Notebook使用的Python内核环境，和!pip install调用的pip所属环境不一致的问题——虚拟机里通常存在多个Python版本/隔离环境，导致安装的包没出现在Jupyter当前内核的环境中。

解决方案 #

方案一：用虚拟环境（推荐，虚拟机完全适用） #

虚拟环境是隔离项目依赖的标准方案，在虚拟机中使用完全没问题，步骤如下：

1. 退出Jupyter，回到SSH终端，创建虚拟环境：

   python3 -m venv torch_env
   

2. 激活虚拟环境：

   source torch_env/bin/activate
   

3. 安装torch和Jupyter内核依赖：

   pip install torch ipykernel
   

4. 将虚拟环境注册为Jupyter内核：

   python -m ipykernel install --user --name=torch_env
   

5. 重启Jupyter Notebook服务器，打开Notebook后，在顶部菜单栏Kernel > Change kernel里选择torch_env，再执行import torch即可。

方案二：直接匹配Jupyter内核的pip #

如果不想创建虚拟环境，可在Jupyter单元格里直接用当前内核对应的Python来安装：

import sys
!{sys.executable} -m pip install torch

安装完成后重新执行import torch即可。

环境不匹配的验证方法 #

在Jupyter单元格里运行以下代码，查看路径是否一致：

!which pip
import sys
print(sys.executable)

如果输出的pip路径和Python可执行文件路径不对应，就说明环境不一致，上述方案均可解决。

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