主要介绍了pytorch多GPU并行运算的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

Pytorch多GPU运行 设置可用GPU环境变量。例如，使用0号和1号GPU’ os.environ[CUDA_VISIBLE_DEVICES] = '0,1' 设置模型参数放置到多个GPU上。在pytorch1.0之后的版本中，多GPU运行变得十分方便，先将模型的参数设置并行 if torch.cuda.device_count() > 1: print(Let's use, torch.cuda.device_count(), GPUs!)