Hi,

I found this https://github.com/pytorch/examples/pull/73/files to set an env variable to choose GPU.
Is there a way to manually choose GPU programmatically in code? Thanks.