Pytorch 并行

https://github.com/pytorch/serve

1.介绍 pytorch并行处理是指将模型和相应的数据放在多个GPU机器上运行。 假如有4台GPU机器，那么会在每一个GPU机器上拷贝一个模型，然后把数据平均分成四份（若batch_size除不尽，会自动适配） 2.代码 self.device = torch.device("cuda" if to