监控GPU使用情况;提升GPU利用率;Pytorch解决cuda out of memory

1、watch -n 0.1 nvidia-smi

Pwr:Usage/Cap(48w/250w)表示显卡功率;中间那一栏,表示显存的已占用/总共的大小;Perf表示性能状态,从P0到P12,P0表示最大性能,P12表示状态最小性能(未工作时为P0);FAN表示风扇转速;persistence-M:持续模式的状态,持续模式虽然耗能大,但是在新的GPU应用启动时,花费的时间更少;Disp.A:display active,表示GPU的显示是否初始化;volatile GPU-util:浮动的GPU利用率; compute M:计算模式。

 

2、cuda out of memory

在网络中,存在一个generator和3个discriminator,loss是四者的和,在训练generator时,将discriminator的计算放在with torch.no_grad():语句下;同样,在训练discriminator时,将generator的计算放在with torch.no_grad()语句下。避免多余的反向传播

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值