Python 和环境变量不起作用

Question

在 Ubuntu 上的 Docker 容器中，我正在运行一个调用python3 my_script_file.py的.sh文件。 在那个文件中，我这样做：

commands = []
commands.append("export OMP_NUM_THREADS=4")
commands.append("echo $OMP_NUM_THREADS")

然后我执行命令数组：

def execute_commands_list(commands, no_bash_script = False):
    for command in commands:
        if not no_bash_script and gc.output_to_bash_script:
            write_command_to_bash_script(command)
        else:
            print("     [  " + command + "  ]")

            ret = subprocess.call(command, shell=True, executable="/bin/bash")

            assert ret==0 or ret==2, "bash return code error: " + str(ret) + " " + command

这是输出：

     [  export OMP_NUM_THREADS=4  ]
     [  echo $OMP_NUM_THREADS  ]

（这是一个空行）。 我自己执行后尝试在终端中运行 echo $OMP_NUM_THREADS ，但一无所获。 如果我手动运行这两个命令，则会设置 env var。

（我也试过 os.environ，但也失败了）

我错过了什么？

（作为参考，.sh 文件是:)

#!/bin/bash -p

python3 my_script_file.py

Answer 1

这可能是因为export的范围仅限于它正在运行的当前 shell。 这意味着只有subprocess运行的 shell，这个环境变量才可见。

如果你想让这个环境变量全局公开，并且你是从某种 Linux 机器上运行的，我建议将它以KEY=VALUE的格式写入/etc/environment 。

例如在你的情况下，而不是

export OMP_NUM_THREADS=4

替换为：

echo OMP_NUM_THREADS=4 >> /etc/environment

笔记！ 由于environment位于/etc下，很有可能将其权限设置为仅允许写入 sudo 用户，因此请确保将其包含在subprocess调用中。

您可以使用以下内容：

echo 'OMP_NUM_THREADS=4' | sudo tee -a /etc/environment > /dev/null

tee将文本打印到标准输出。 为了使其静音，使其行为更类似于 shell 附加 ( >> )，请将 stdout 路由到/dev/null 。