這裡是 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可在運行時這樣寫。

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！