os.environ环境变量以及GPU的指定以及内存占用

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'

你可能感兴趣的:(python)