Specify GPU

[CerebralSeed]

Could something be added to specify the GPU to use?

[vadim epstein]

CUDA_VISIBLE_DEVICES environment variable perfectly controls this.
you can set it from command line: set CUDA_VISIBLE_DEVICES=0
or from python itself:

import os
os.environ['CUDA_VISIBLE_DEVICES']='0'

[CerebralSeed]

Okay, got it. I will give that a try.