Tensorflow 指定 GPU 和修改日志等级

2020-06-22  本文已影响0人  momo1023

指定 GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '0'

TF 日志等级管理

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
上一篇下一篇

猜你喜欢

热点阅读