search

tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts
MIT License
1k stars 89 forks source link

When the batch size is set to 2, how much gpu memory is required at least. #70

Closed wangqinxuan closed 5 months ago

wangqinxuan commented 5 months ago

My 3090 graphics card has 24G of memory but it’s still not enough. So how much gpu memory is required at least, when the batch size is set to 2.

tianrun-chen commented 5 months ago

Hello! We suggest using A100, the application need at least 60Gb of VRAM due to the size of original SAM. But you can try use ViT-L with a 24Gb memory card.