指定TensorFlow使用哪一个GPU

来源:互联网 发布:虚拟币交易所源码 编辑:程序博客网 时间:2024/05/18 21:50

    在机器学习中,如果不设置GPU使用情况,会默认使用第一块GPU,而且会完全占用所有GPU资源。使用下面设置就能轻松解决问题。

1
2
3
$ import os
$ os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
$ os.environ["CUDA_VISIBLE_DEVICES"]="0"

原创粉丝点击