首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >问答 | 前向推理的时候 Yolov3 怎么指定 GPU?

问答 | 前向推理的时候 Yolov3 怎么指定 GPU?

作者头像
AI研习社
发布2018-09-25 15:57:31
1.4K0
发布2018-09-25 15:57:31
举报
文章被收录于专栏:AI研习社AI研习社
社长为你推荐来自 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可在运行时这样写。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2018-09-08,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 AI研习社 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 话不多说,直接上题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档