标签：mode 训练 parallel article detail 增加 href blog 单机
只需要在model定义处增加下面一行：
model = model.to(device) # device为0号
model = torch.nn.DataParallel(model)
标签：mode 训练 parallel article detail 增加 href blog 单机
原文地址：https://www.cnblogs.com/huanxifan/p/13187656.html
