跑模型——常用模块之os.environ

前言
在 python 中通过 os.environ 可以获取有关系统的各种信息

1、简介

os.environ 是一个字典，是环境变量的字典，通过 os.environ 我们可以获取环境变量，例如“HOME” 是这个字典里的一个键，如果有这个键，返回对应的值，如果没有，则返回 none

1.1 常见 key 字段

Windows：

os.environ['HOMEPATH']:当前用户主目录。
os.environ['TEMP']:临时目录路径。
os.environ["PATHEXT"]:可执行文件。
os.environ['SYSTEMROOT']:系统主目录。
os.environ['LOGONSERVER']:机器名。
os.environ['PROMPT']:设置提示符。

Linux：

os.environ['USER']:当前使用用户。
os.environ['LC_COLLATE']:路径扩展的结果排序时的字母顺序。
os.environ['SHELL']:使用shell的类型。
os.environ['LAN']:使用的语言。
os.environ['SSH_AUTH_SOCK']:ssh的执行路径。

1.2 os.environ.get() 用法

os.environ 是一个字典，是环境变量的字典，可以通过 get 方法获取键对应的值（注意 os.environ 的类型并不是 <class ‘dict’>，不是所有字典的函数都能用）

os.environ.get() 是 python 中 os 模块获取环境变量的一个方法，如果有这个键，返回对应的值，如果没有，则返回 none

import os
print(os.environ.get("HOME"))

也可以设置默认值，当键存在时返回对应的值，不存在时，返回默认值

print(os.environ.get("HOME", "default"))	#环境变量HOME不存在，返回	default

1.3 环境变量用法总结–设置、修改、获取、删除、判断

①设置系统环境变量

os.environ['环境变量名称']='环境变量值' #其中key和value均为string类型
os.putenv('环境变量名称', '环境变量值')
os.environ.setdefault('环境变量名称', '环境变量值')

②修改系统环境变量

os.environ['环境变量名称']='新环境变量值'

③获取系统环境变量

os.environ['环境变量名称']
os.getenv('环境变量名称')
os.environ.get('环境变量名称', '默认值')	#默认值可给可不给，环境变量不存在返回默认值

④删除系统环境变量

del os.environ['环境变量名称']
del(os.environ['环境变量名称'])

⑤判断系统环境变量是否存在

'环境变量值' in os.environ   # 存在返回 True，不存在返回 False

如：

    #判断当前程序是否在SLURM集群上运行
    if "SLURM_JOB_ID" in os.environ:
        #获取进程ID
        rank = int(os.environ["SLURM_PROCID"])
        #任务总数
        world_size = int(os.environ["SLURM_NTASKS"])
        #节点列表
        node_list = os.environ["SLURM_NODELIST"]

原文链接：https://blog.csdn.net/happyjacob/article/details/109279118