多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...
原创
2019-03-20 17:22:38 ·
2104 阅读 ·
0 评论