美文网首页
约束 TensorFlow 只在 GPU 上运行

约束 TensorFlow 只在 GPU 上运行

作者: 不做秃顶的程序猿 | 来源:发表于2019-01-08 14:35 被阅读0次

    显卡资源有限,当有显卡被占用时可以让 TensorFlow 在 CPU 上继续执行任务

    只需要添加如下代码即可

    import os
    os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
    os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
    

    该方法是让 tf 无法调用 cuda,降级到使用 CPU 资源进行计算

    相关文章

      网友评论

          本文标题:约束 TensorFlow 只在 GPU 上运行

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