pytorch单机多卡训练
训练
只需要在model定义处增加下面一行：
model = model.to(device) # device为0号
model = torch.nn.DataParallel(model)
载入模型
- 如果是多GPU载入，没有问题
- 如果训练时是多GPU，但是测试时是单GPU，会出现报错 解决办法
pytorch单机多卡训练
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。