Question

我在批处理中运行python脚本。我想导出批处理变量并在python命令行中加载它，如下所示：

#!/bin/bash -l
#SBATCH -n 32
#SBATCH -N 1
#SBATCH -t 24:00:00

export OMP_NUM_THREADS=1
APRUN_OPTIONS="-n 32 -d 1 -cc none"

python script.py

然后，在python脚本中，当我使用我提交它时：

aprun_opts = os.environ['APRUN_OPTIONS']
command = "aprun " + aprun_opts + " command_execute"

使用os.system（命令）

到目前为止，我收到错误：

Traceback (most recent call last):
  File "script.py", line 11, in <module>
    aprun_opts = os.environ['APRUN_OPTIONS']
  File "/usr/lib64/python2.6/UserDict.py", line 22, in __getitem__   raise KeyError(key)
KeyError: 'APRUN_OPTIONS'

有人能看到问题吗？谢谢！

Answer 1

如果您的脚本中不需要APRUN_OPTIONS，则只需执行

即可

APRUN_OPTIONS="-n 32 -d 1" python script.py

请注意，它们位于同一行，并且没有export。

在python脚本中加载导出的环境变量

1 个答案: