Tensorflow 指定 GPU 和修改日志等级

指定 GPU

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

TF 日志等级管理

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

你可能感兴趣的:(Tensorflow 指定 GPU 和修改日志等级)