create a new flag init_gpu_device that select the gpu to use, but don't automatically move computation to it.