PAI-DSW中对齐NoteBook和命令行的Python环境

我在命令行使用 pip 安装了neo4j，但是在NoteBook中却无法import成功，问了Chat-GPT才知道可能是NoteBook和命令行的Python不一致造成的。下面将介绍如何验证NoteBook和命令行的Python环境是否一致，以及如何将NoteBook的Python环境修改成命令行的Python环境。

本文涉及的代码几乎都是GPT-4o生成。它们对于其他的云服务器应该也是适用的。

一、验证NoteBook和命令行的Python环境是否一致

1. 验证NoteBook

import sys
print(sys.executable)

输出如下：

2. 验证命令行

python -c "import sys; print(sys.executable)"

输出如下：

可知两者的Python环境不一样。

二、将NoteBook的Python环境对齐成命令行的Python环境

1. 安装ipykernel
   首先，确保在/root/anaconda3/bin/python环境中安装了ipykernel。在命令行中运行以下命令：

/root/anaconda3/bin/python -m pip install ipykernel

2. 创建新的内核
   使用ipykernel创建一个新的Jupyter内核，指向/root/anaconda3/bin/python的环境。在命令行中运行以下命令：

/root/anaconda3/bin/python -m ipykernel install --user --name anaconda3 --display-name "Python (Anaconda3)"

这里，–name anaconda3是内核的名称，–display-name "Python (Anaconda3)"是显示在Jupyter Notebook中的名字。您可以根据需要修改这些名称。

创建成功会返回如下结果：

3. 选择新内核
   打开Jupyter Notebook，在顶部菜单栏中选择 Kernel -> Change kernel，然后选择刚刚创建的“Python (Anaconda3)”内核。如下图：
   

4. 验证内核
   确认内核切换成功后，您可以在Notebook中运行以下代码来验证环境：

import sys
print(sys.executable)

写在最后：
我认为在最开始NoteBook和命令行的Python环境是一致的，只是我前面在部署LLaMA-Factory时进行了如下操作将命令行的Python环境改成了/root/anaconda3/bin/python。具体详见 LLaMA-Factory私有化部署 – PAI-DSW