8张A800(80G)微调Cogagent依然报错： CUDA out of memory

[GuoXu-booo]

System Info / 系統信息

torch 2.0.1+cu118 torchaudio 2.0.2+cu118 torchvision 0.15.2+cu118 cuda 11.8

Who can help? / 谁可以帮助到您？

Information / 问题信息

• [X] The official example scripts / 官方的示例脚本
• [ ] My own modified scripts / 我自己修改的脚本和任务

Reproduction / 复现过程

按照官方提供的脚本执行：bash finetune_cogagent_lora.sh 模型文件使用的是sat模型权重。

Expected behavior / 期待表现

官方提到微调需要的配置是4张A100即可