pytorch中多GPU训练

如果指定在某个GPU（如0号GPU）上训练，则只需要添加： import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 如果需要同时使用多个GPU，则参考：https://oldpan.me/archives/pytorch-to-use-multiple-gpus Pytorch中多GPU训练指北   前言 在数据越来越多的时代，随着模型规模参数的增多，

>>阅读原文<<