search

digitalbrain79 / pyyolo

Simple python wrapper for YOLO.
126 stars 65 forks source link

support different GPU ID #12

Open cjjsaic opened 7 years ago

cjjsaic commented 7 years ago

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 commented 7 years ago

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

cjjsaic commented 7 years ago

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 commented 7 years ago

I will add parameter to select GPUID later.