How to change the default device of GPU? device_ids[0]

[Annie_Lee]

Many thanks! It’s really helpful!