LLaVA问题：Deepspeed无法检测到可编辑安装的Python模块

核心问题根源 #

Deepspeed在shell脚本中执行时，大概率没加载conda环境的Python路径，或者系统默认Python的优先级高于conda环境的Python，导致找不到通过pip install -e .安装在当前conda环境的llava模块。不同系统的shell环境配置差异（比如环境变量加载顺序、默认shell类型），是导致这个问题在部分系统出现的关键。

可行解决方案 #

1. 在shell脚本开头显式激活conda环境 #

确保脚本先切换到目标conda环境，避免Deepspeed调用系统默认Python：

#!/bin/bash
# 替换成你的conda环境名称
conda activate your_env_name
# 执行deepspeed命令
deepspeed main.py --args

2. 直接指定Deepspeed使用conda环境的Python路径 #

明确指定conda环境内的Python解释器路径，强制Deepspeed使用正确环境：

# 替换成你的conda环境Python实际路径，比如~/miniconda3/envs/your_env/bin/python
deepspeed --python ~/miniconda3/envs/your_env/bin/python main.py --args

3. 手动设置PYTHONPATH环境变量 #

如果conda环境的路径没被正确加入PYTHONPATH，可在脚本中手动添加llava项目路径：

#!/bin/bash
conda activate your_env_name
# 替换成你的llava项目根目录路径
export PYTHONPATH="/path/to/your/llava/project:$PYTHONPATH"
deepspeed main.py --args

注：pip install -e .会在conda环境的site-packages目录生成llava的软链接，也可以直接将该site-packages路径加入PYTHONPATH。

4. 排查不同系统的shell配置差异 #

若某系统可用另一系统不可用，重点检查：

• 目标系统的默认shell类型：conda默认在bashrc中配置环境变量，如果是zsh，需要手动在zshrc中添加conda初始化代码
• 确认conda环境在非登录shell中能否正常加载：如果是首次使用，可能需要执行conda init完成shell初始化

环境验证方法 #

在shell脚本中先执行以下命令，确认环境配置正确：

conda activate your_env_name
which python
python -c "import llava; print(llava.__file__)"

如果这两步都能正常输出llava的安装路径，再执行Deepspeed命令即可确保环境一致。

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