How to bind net on specific GPU id
crqcrq001 opened this issue · 3 comments
crqcrq001 commented
how to set gpu id ?
Shixiaowei02 commented
您好，对于 Nvidia GPU，最快的办法是设置环境变量 CUDA_VISIBLE_DEVICES。
crqcrq001 commented
分线程绑定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 commented
@crqcrq001 好的，谢谢！抱歉没及时回答，若有急需解决的问题也可以 email 至 anakin@baidu.com。谢谢！