How to avoid : CUDA out of memory. Tried to allocate 86.23 GiB error? #316

Closed Unanswered

automataIA asked this question in Q&A

automataIA
Sep 19, 2024

Hi,
I try to make a mask of sentinel satellite images. Are 10'000X10'000 pixel dimension (about 200 MB).
How to work with this type of image? Advice?

This is my code(mask.generate give me the error).

sam = SamGeo(model_type="vit_h")  
# Perform segmentation on the GeoTIFF image
masks = sam.generate(input_tiff, output=None, foreground=True, unique=True)

Replies: 0 comments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment