1、设置环境变量,放在代码最前面的(但是要先导入os包)。
os.environ[‘环境变量名称’]=‘环境变量值’ #其中key和value均为string类型。例如:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
注意
如果"CUDA_VISIBLE_DEVICES" 书写错误,也不报错,自动选用第一个显卡
"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信息
1、log信息共有四个等级,按重要性递增为:
INFO(通知)<WARNING(警告)<ERROR(错误)<FATAL(致命的);
2、值的含义:不同值设置的是基础log信息(base_loging),运行时会输出base等级及其之上(更为严重)的信息。具体如下:
注意:
1、“0”为默认值,输出所有信息
2、设置为3时,不是说任何信息都不输出,ERROR之上还有FATAL