How to bind net on specific GPU id

[crqcrq001]

how to set gpu id ?

[Shixiaowei02]

您好，对于 Nvidia GPU，最快的办法是设置环境变量 CUDA_VISIBLE_DEVICES。

[crqcrq001]

分线程绑定gpu我也搞定了，我的做法是在初始化的时候改traget wrapper device 然后在每次前向的时候把输入也绑到指定gpu上，我也实测过 多个gpu确实有同时占用且都跑的挺满的，结果令人满意 在 2018-09-21 11:23:54，"石晓伟" <notifications@github.com> 写道： 您好，对于 Nvidia GPU，最快的办法是设置环境变量 CUDA_VISIBLE_DEVICES。 — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.

[Shixiaowei02]

@crqcrq001 好的，谢谢！抱歉没及时回答，若有急需解决的问题也可以 email 至 anakin@baidu.com。谢谢！