pytorch多GPU使用的方法model=nn.DataParallel(model).cuda()

单GPU：

import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 1 2

多GPU：

device_ids = [0,1,2,3]