Python为子进程设置环境变量

1. 问题描述

安装软件，在.bashrc或者.zshrc中配置软件的环境变量之后，python代码的子进程仍然无法使用该环境变量。

例如，在无sudo的情况下，安装gcc至/home/usr/app。在配置好环境变量之后，gcc已经可以正常使用，但是运行的python代码中涉及到子进程，子进程继续无法使用gcc。

2. 解决方法

需要手动在代码中添加环境变量。比方说，如果gcc的使用，需要倒入环境变量export PATH=$HOME/app/bin:$HOME/app/lib64:$PATH 和 export LD_LIBRARY_PATH=$HOME/app/lib/:$LD_LIBRARY_PATH的话，那么python代码中也要相应地设置这几个环境变量：

import os

# get current env
my_env = os.environ.copy()  
# append the env of target app
my_env["PATH"] = "/home/usr/app/bin:" +\
                "/home/usr/app/lib64:" +\
                my_env["PATH"]
my_env["LD_LIBRARY_PATH"] = "/home/usr/app/lib/:" +\
                my_env["LD_LIBRARY_PATH"]
# update the env                
os.environ.update(my_env)

这种方法适用于任何一种，环境变量在进程中无法生效的情况, 不單單針對於子進程。

3. 参考：

https://github.com/zhanghang1989/PyTorch-Encoding/issues/167