Tensorflow

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值