tensorflow 多GPU训练时, Memory-Usage占满而GPU-Util为0%的问题

      由于刚开始接触多gup训练,按照网上的帖子改写代码后,发现训练程序依旧很慢(用的是4gpu),用nvidia-smi检查GPU运行情况,发现Memory-Usage占满而GPU-Util为0%。

      上网一搜还真有不少人遇到同样的问题,还以为是自己哪里没设置好。画了一天左右时间在网上找解决方法,网上的方法都没用。

      后来用watch nvidia-smi命令,实时观察gpu运行状况发现,大部分时间4个gpu的GPU-Util为0%,突然有一个时间4个gpu的GPU-Util为35%了。这时才注意到自己写的多gpu训练部分的程序已经成功运行了的。只是大部分时间在cpu上运行,极少部分时间在gpu上运行。于是仔细查看代码,发现大部分时间花在数据增强部分的程序上了。把数据增强部分代码去掉后,GPU-Util不再为0%了。

总结:

1.要用watch nvidia-smi命令查看GPU运行状况。

2.如果想要利用gpu提高训练效率就不要进行数据增强操作,直接拿大数据集来用。

 

 

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值