开发者社区 > ModelScope模型即服务 > 多模态 > 正文

OFA文生图模型,显存空间不足的问题如何解决

当我使用OFA文生图模型使用CPU时,模型写出速度非常缓慢,当我更新torch为GPU版本后,会报错torch.cuda.OutOfMemoryError: CUDA out of memory 显存的内存空间不够的问题,请问这种问题如何去解决? GPU至少需要多大的内存空间 image.png

我的torch版本为1.13.0 torch.cuda.is_available() # TRUE 显卡为GTX1060 6G版本

展开
收起
aliyun9665825026 2022-11-24 11:01:17 1737 0
1 条回答
写回答
取消 提交回答
  • 您好,目前CPU版本确实非常慢,GPU需要16G以上显存才能得到一个相对较好的效果。这主要是因为图片本身相当于极长的序列,生成长序列需要很大的显存。

    2022-11-25 11:38:49
    赞同 展开评论 打赏

包含图像描述、文本生成图片、版面分析、多模态表征、视觉问答、文档理解等多个领域

热门讨论

热门文章

相关电子书

更多
通义万相:视觉生成大模型的进化与应用 立即下载
低代码开发师(初级)实战教程 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载