Pytorch

1.pytorch多gpu

import torch.nn as nn

model=nn.DataParallel(model,device_ids=[0,1,2]) # multi-GPU