1.可能是GPU被其他程序占用内存,需要先清理GPU内存:https://blog.csdn.net/weixin_43398590/article/details/105383173
2.代码造成的GPU内存负载:https://blog.csdn.net/c2a2o2/article/details/90321614?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.control&dist_request_id=b13ff9c8-213e-4da5-abb0-fff34eede034&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.control、https://blog.csdn.net/weixin_36670529/article/details/106041652。
3.batch_size过大造成,调整batch_size。