训练模型时内存占用率的问题

asscalar()这个函数不是实现同步计算,降低显存占有率提高计算性能的吗,怎么我程序到这里GPU显存占有率突然升高,然后就out of memory?求大神解答

解决out of memory
1 .打开任务管理器,找到你训练的gpu,找到专用gpu内存利用率,这就是你的显卡内存,简称显存,爆内存也就是这里的内存不够训练时数据的存放,训练完之后专用内存利用率还是不会降下来,目前已知方法是重启重启软件。