support different GPU ID

[cjjsaic]

trafficstars

hi, @thomaspark-pkj
@jasrk11 @oulutan @rayhou0710

Try to debug the code with different GPUID instead of default 0.

I try to update the wrapper but failed. Could you help provide some hint? Thanks

[rakeshjasti]

You can do that by using the command cuda_set_device(gpu_id) before you parse your cfg file

[cjjsaic]

Hi, @jasrk11,

i tried this on python wrapper. it seemed working.

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "1"

[digitalbrain79]

I will add parameter to select GPUID later.