pytorch训练中指定GPU

有两种方式：

1.代码中指定GPU：

import os
os.environ['CUDA_VISIBLE_DEVICE'] = 1

2.训练的时候通过命令进行指定：

CUDA_VISIBLE_DEVICES=1