When using GPU 3070, there may be insufficient video memory.

vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything (SAM+SAM2), MobileSAM!!

https://anylabeling.nrl.ai

GNU General Public License v3.0

2.3k stars 240 forks source link

When using GPU 3070, there may be insufficient video memory. #54

Closed GitHubChrischen closed 1 year ago

GitHubChrischen commented 1 year ago

Please do not infer all images in the folder at once. set or automatically pre-infer a few images according to the current video card model to ensure that the increase of video memory is within an acceptable range.

vietanhdev commented 1 year ago

Currently, the model only runs on 1 image once. Can you capture the error messages?

vietanhdev commented 1 year ago

Documentation for GPU support: https://anylabeling.com/docs/gpu.