前向推理的时候 Yolov3 怎么指定 GPU？

这里是 AI 研习社，我们的问答版块已经正式推出了！欢迎大家来多多交流~

https://club.leiphone.com/page/question

社长为你推荐来自 AI 研习社问答社区的精华问答。如有你也有问题，欢迎进社区提问。

话不多说，直接上题

问：前向推理的时候 Yolov3 怎么指定 GPU？

来自社友的回答

@邱忠喜：

os.environ[“CUDA_VISIBLE_DEVICES“]=INDEX index用GPUid代替id从前0开始，

也可在python文件中指定用os.environ[“CUDA_DEVICE_ORDER“]=“PCI_BUS_ID“] os.environ[“-，

其中index用可以用 nvidia-smi查看比如0，

CUDA_VISIBLE_DEVICES=index python xxx.py，

使用cuda可以指定可用的GPU可在运行时这样写。