计算余弦相似度时显示内存不足

[cfxdfk]

作者你好 我使用的配置是官方要求的3090*2 每张24g 想要训练自己的数据集 但在训练时却显示显存不足 关键点显示在余弦相似度的计算时预分配10g内存但在加载完imagebind和vicuna模型后显存已不足 目前尝试的方法有降低批次 删除无关类别 都未起到效果