os.environ输出的是一个字典包含key和value;
用Python Shell设置或获取环境变量的方法:
一、设置系统环境变量
1、os.environ['环境变量名称']='环境变量值' #其中key和value均为string类型
2、os.putenv('环境变量名称', '环境变量值')
GPU
import os os.environ["CUDA_VISIBLE_DEVICES"]=‘6‘’,‘7’
tensorflow中设置占用每个GPU的比例
(1)定量设置显存
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) 这样运行TensorFlow程序时,每个使用的GPU中,占用的显存都不超过总显存的0.7.
(2)按需设置显存
gpu_options = tf.GPUOptions(allow_growth=True) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) 这样设置以后,程序就会按需占用GPU显存。
二、获取系统环境变量
1、os.environ['环境变量名称']
2、os.getenv('环境变量名称')
例如
“if "BATCH_CONFIG_INI" in os.environ:” 判断环境变量BATCH_CONFIG_INI的值有没有定义
使用os.environ.keys()可以查看所有的Key
三、key字段详解
linux:
os.environ['HOME']:用户地址。’/home/zhu'
os.environ['USER']:当前使用用户。‘zhu'
os.environ['SHELL']:使用shell的类型。'/bin/bash'
os.environ['LANGUAGE']:使用的语言。'zh_CN:en'
os.environ['SSH_AUTH_SOCK']:ssh的执行路径。'/run/user/1000/keyring/ssh'