Tensorflow 多GPU下,指定使用某一块GPU

来源:互联网 发布:c语言中如何打开文件 编辑:程序博客网 时间:2024/05/28 15:06

如果机器中有多块GPU,tensorflow会默认吃掉所有能用的显存,
如果实验室多人公用一台服务器,希望指定使用特定某块GPU。
可以在文件开头加入如下代码:

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

指定使用第二块GPU(从0开始)

5 0