开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

modelScope用显存推理应该加什么参数?

7002272fd7f13e0d073c3d4a729ba685.png
modelScope用显存推理应该加什么参数?

展开
收起
小小爱吃香菜 2024-01-09 20:39:27 55 0
2 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    在使用ModelScope进行显存推理时,如果遇到显存使用量过高的问题,您可以尝试以下方法:

    1. 减小模型尺寸:您可以选择更小的模型,这样可以减少模型参数和计算量,从而降低显存的需求。例如,对于一些需要大量显存的大模型,可以考虑使用轻量级模型。

    2. 使用分布式推理:如果您有可用的多个GPU,可以尝试使用分布式推理来将负载分散到多个GPU上,从而降低单个GPU的显存压力。

    在具体的ModelScope操作中,可以在pipeline函数中添加gpu_mem参数来限制显存的使用。例如,gpu_mem=8000表示限制显存使用量为8GB。但请注意,设置的值不能超过实际显存的大小。

    2024-01-13 19:32:21
    赞同 展开评论 打赏
  • pipeline初始化时device='gpu'。但一般会自动用显卡。此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-01-09 23:14:04
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载