sudo运行Python时如何调用非默认指定目录的第三方包

报错根因 #

sudo执行命令时默认会重置用户环境变量，root身份运行的Python解释器的包搜索路径（sys.path）默认不包含普通用户的本地第三方包目录（即提到的/home/user1/.local/lib/python3.8/site-packages），因此无法找到用户目录下安装的torch包。

以下是三种经过验证的可行方案，按推荐优先级排序： #

方案1：单次执行临时传入包路径（最稳妥，无永久配置修改） #

执行sudo命令时手动将用户本地包路径追加到PYTHONPATH环境变量，并传递给sudo启动的进程，仅对当前命令生效，不会污染全局环境：

sudo PYTHONPATH="/home/user1/.local/lib/python3.8/site-packages:$PYTHONPATH" python3 toy_example.py

如果需要适配不同Python版本的本地路径，可以先动态获取当前用户的site-packages路径再执行，避免硬编码路径出错：

USER_PYTHON_SITE=$(python3 -c "import site; print(site.getusersitepackages())")
sudo PYTHONPATH="$USER_PYTHON_SITE:$PYTHONPATH" python3 toy_example.py

注意：执行前请确认root用户对该包目录有读取权限，若有权限问题可提前执行sudo chmod -R o+rX /home/user1/.local/lib/python3.8/site-packages开放读取权限。

方案2：修改sudo配置永久保留PYTHONPATH（适合高频使用场景） #

如果需要频繁用sudo跑依赖用户本地包的Python脚本，可以修改sudo配置，让sudo执行时自动继承当前用户的PYTHONPATH变量：

1. 执行sudo visudo打开sudo配置文件（该命令会自动校验配置语法，避免手动编辑文件出错导致sudo不可用）
2. 找到配置中Defaults env_reset的行，在其下方新增一行配置：

   Defaults env_keep += "PYTHONPATH"
   

3. 保存退出后，编辑当前用户的shell配置文件（比如bash对应~/.bashrc，zsh对应~/.zshrc），在末尾追加路径配置：

   export PYTHONPATH="/home/user1/.local/lib/python3.8/site-packages:$PYTHONPATH"
   

4. 执行source ~/.bashrc（对应你自己的shell配置文件）让配置生效，后续直接执行sudo python3 xxx.py即可正常识别用户本地安装的包。

方案3：脚本内部动态添加搜索路径（适合脚本分发，不依赖外部环境） #

如果不想修改任何系统或shell配置，可以直接在Python脚本的最开头、导入第三方包之前，手动将用户包路径加入Python的搜索路径列表：

import sys
# 注意这行要写在import torch之前
sys.path.append("/home/user1/.local/lib/python3.8/site-packages")

import torch
# 后续业务逻辑

避坑提示 #

• 不要为了解决导入问题直接执行sudo pip3 install torch将包装到root全局环境，可能覆盖系统Python的默认依赖，导致系统自带的Python工具（比如apt相关的Python脚本）运行异常。
• 不要随意修改/home/user1/.local目录的所有者为root，会导致普通用户后续执行pip本地装包时出现权限错误。

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