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

最新推荐文章于 2024-04-04 22:01:54 发布

noirblack

最新推荐文章于 2024-04-04 22:01:54 发布

阅读量1.9w

点赞数 9

分类专栏： Tensorflow python 文章标签： Tensorflow

本文链接：https://blog.csdn.net/noirblack/article/details/70792550

版权

9 篇文章 1 订阅

订阅专栏

5 篇文章 0 订阅

订阅专栏

如果机器中有多块GPU，tensorflow会默认吃掉所有能用的显存，
如果实验室多人公用一台服务器，希望指定使用特定某块GPU。

可以在文件开头加入如下代码：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"       # 使用第二块GPU（从0开始）

也可以制定使用某几块GPU

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"    # 使用第一, 三块GPU

禁用GPU

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

关注