关于tensorflow中os.environ的设置

本帖旨在记录自己平时遇到的各种关于os.environ的设置，不定期更新
 
0.使用方法：设置环境变量，自然是要放在代码最前面的（但是要先导入os包）。例如下面：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

 
1."CUDA_VISIBLE_DEVICES"：自定义要使用的GPU（tensorflow中默认是全部使用的）
注意：GPU一般是从0开始编号的，可以通过nvidia-smi查看，如下图：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"  #仅使用GPU2（也就是第3块GPU）
os.environ["CUDA_VISIBLE_DEVICES"] = "2, 3"  #也可以指定多块
os.environ["CUDA_VISIBLE_DEVICES"] = ""  #不使用GPU

 
2."TF_CPP_MIN_LOG_LEVEL"：设置log级别（也就是程序运行时打印什么信息）

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='0'  #默认值，打印所有信息
os.environ['TF_CPP_MIN_LOG_LEVEL']='1'  #屏蔽INFO信息
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'  #屏蔽INFO与WARNING信息
os.environ['TF_CPP_MIN_LOG_LEVEL']='3'  #屏蔽INFO, WARING, ERROR信息

实际上就是有一个优先级（从左到右严重性递增）：INFO<WARING<ERROR<FATAL
注意：设置为3不是说什么信息都不打印了，ERROR之上还有FATAL（这个单词是致命的意思）