Demo for codon GPU set_device

[qinwf]

I build codon from source with GPU, and how to use GPU set_device like cuda code.

Thanks.

[arshajii]

This is currently exposed in the API (although it should be), but you can use the CUDA_VISIBLE_DEVICES environment variable to control which devices gets used.