【环境配置】Pytorch指定使用GPU编号训练数据