Adds an option to select device ID when using CUDA

[hassyGo]

Hi,

thank you for sharing your easy-to-use code.

I think it is helpful to add an option to specify a device ID when using the --cuda option.

Best,