(低显存)如何实现segment anything等大模型本地部署

本文以SAM为例

前提:本人电脑比较老旧,显存不够只有4g,故出现“CUDA out of memory”

1.如何查看显存

win+r,cmd里输入dxdiag,诊断工具点击确认,左上角点击显示,找到显示内存(VRAM)

2.如何扩充显存

第一步:先把模型的权重调到最小,显存本来就少的用base就行了,附带一句官方文档给的6.9g显存才能跑,显卡满足的就可以不用往下看了

第二步:英伟达有个功能可以实现共享内存,把cpu拉过来当gpu用,官方文档System Memory Fallback for Stable Diffusion | NVIDIA

546.01后的版本提供这个功能的开关,是默认开启的,然后就是最重要的一步

在联想电脑管家里把独显模式关了,开启混合模式

这里是需要重启电脑生效

重启完就可以跑了

  • 5
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值