pytorch多个显卡并行训练 RuntimeError: Caught RuntimeError in replica 0 on device 0.

今天复现并改动一份代码的时候,发生了该错误:

RuntimeError: Caught RuntimeError in replica 0 on device 0.

原代码可以再单卡上面正常运行。但因为自己的需要改成多GPU并行加载模型,然后就发生上述错误,并且还提示没有model.drop_path_prob这个属性!问题出现的原因可以看看这篇:https://blog.csdn.net/qq_30614451/article/details/106766551   简单来说:就是 网络模型的某个参数复制分配到 不同的GPU的时候,部分参数始终在GPU_0上

解决方案

提前声明  需要用到的模型参数,如下图最后一行

我的self.drop_path_prob这个属性,在__init__()函数的后面有直接使用,因为python中不需要特别预定义也可以直接使用变量。

但是torch.nn.DataParallel并不认可这种没有事先声明的变量,所以其他GPU没有拷贝这个drop_path_prob模型属性变量。

 

仅为个人拙劣的解决方案,做个笔记 ~0.0

  • 11
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值