训练时选择GPU

1 在代码中通过GPU index指定device;

2 启动时指定：

CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py