Specify GPU

[Onyoursix]

My box has multiple GPUs, is there a way to specify which GPU I want to process on?

[hbredin]

Setting CUDA_VISIBLE_DEVICES environment variable should do the trick.
See this and this.

[hbredin]

This PR adds a new pipeline.to(device) API. Would you mind testing it?