Open cjjsaic opened 7 years ago
You can do that by using the command cuda_set_device(gpu_id)
before you parse your cfg file
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"
I will add parameter to select GPUID later.
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