多GPU實現同時訓練模型

1.1單GPU： import os os.environ[「CUDA_VISIBLE_DEVICES」] = 「0」 說明：設置單塊GPU編號 1.2多GPU： device_ids = [0,2,4,6] #說明：GPU編號，一塊顯卡集成兩塊GPU，則同一塊卡的編號不能同時出現。 model = nn.DataParallel(model, device_ids=device_ids).cud

>>阅读原文<<