如何根据PCI总线id选择GPU？
如何根据PCI总线id选择GPU？
提问于 2019-11-12 15:18:33
我想知道是否可以根据pci总线id或uuid在Tensorflow和Pytorch中选择GPU？
回答 2
Stack Overflow用户
发布于 2019-11-12 16:47:41
您可以通过设置CUDA_VISIBLE_DEVICES环境变量来执行此操作。假设你想在PCI总线id = 1的GPU上训练一个模型，你需要这样做，
$ CUDA_VISIBLE_DEVICES=1 my_training_code.py
您也可以通过在python代码中使用os.environ设置CUDA_VISIBLE_DEVICES来实现同样的目的
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
...Stack Overflow用户
发布于 2019-11-12 19:43:00
使用tensorflow进行测试
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0" #Select GPU number 0页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/58813757
复制相关文章
相似问题
腾讯云开发者
