使用多块GPU，并行训练数据的pytorch代码

使用多块GPU，并行训练数据的pytorch代码

if torch.cuda.device_count() > 1:
    print ("Let's use {0} GPUs!".format(torch.cuda.device_count()))
    model = nn.DataParallel(model, device_ids=[0,1,2,3])# device_ids指的是你想用哪几块GPU进行训练