Semantic-Segment-Anything icon indicating copy to clipboard operation
Semantic-Segment-Anything copied to clipboard

Published 20 hours ago • verified publisher icon fudan-zvg

out of memory！！！

Open GAOXINYU203 opened this issue 1 year ago • 0 comments

cuda:11.1 torch:1.10 single A6000 命令：python scripts/main_ssa_engine.py --data_dir=/mnt/usb/gxy/dataset_path/images --out_dir=output --world_size=1 --save_img --sam --checkpoint-path=checkpoint-path/sam_vit_h_4b8939.pth 报错：RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 47.54 GiB total capacity; 537.07 MiB already allocated; 12.06 MiB free; 580.00 MiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF help!!!help!!!

GAOXINYU203 avatar Jul 06 '23 09:07 GAOXINYU203