Change gpu id.

[spencerzzzzzzz]

Hello author, I would like to ask how to modify the gpu id when I train and test, because I found that 0 card is used by default every time.