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訓練指北   前言 在數據越來越多的時代，隨着模型規模參數的增多，

>>阅读原文<<