美文网首页
Tensorflow 指定 GPU 和修改日志等级

Tensorflow 指定 GPU 和修改日志等级

作者: momo1023 | 来源:发表于2020-06-22 09:42 被阅读0次

    指定 GPU

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

    TF 日志等级管理

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

    相关文章

      网友评论

          本文标题:Tensorflow 指定 GPU 和修改日志等级

          本文链接:https://www.haomeiwen.com/subject/iratfktx.html