sam-pt
sam-pt copied to clipboard
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.69 GiB already allocated; 0 bytes free; 1.74 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.
Error executing job with overrides: ["frames_path='D:\computer_vision\sam-pt\data\demo_data\demo'", 'query_points_path=null', 'longest_side_length=1024', 'frame_stride=1', 'max_fr
ames=-1']
Traceback (most recent call last):
File "E:\Anaconda3\envs\sam\lib\runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "E:\Anaconda3\envs\sam\lib\runpy.py", line 87, in _run_code
exec(code, run_globals)
File "D:\computer_vision\sam-pt\demo\demo.py", line 372, in
Thread 0x0000524c (most recent call first): File "E:\Anaconda3\envs\sam\lib\threading.py", line 302 in wait File "E:\Anaconda3\envs\sam\lib\site-packages\sentry_sdk_queue.py", line 240 in get File "E:\Anaconda3\envs\sam\lib\site-packages\sentry_sdk\worker.py", line 127 in _target File "E:\Anaconda3\envs\sam\lib\threading.py", line 870 in run File "E:\Anaconda3\envs\sam\lib\threading.py", line 932 in _bootstrap_inner File "E:\Anaconda3\envs\sam\lib\threading.py", line 890 in _bootstrap
Thread 0x000053b0 (most recent call first): File "E:\Anaconda3\envs\sam\lib\site-packages\sentry_sdk\sessions.py", line 117 in _thread File "E:\Anaconda3\envs\sam\lib\threading.py", line 870 in run File "E:\Anaconda3\envs\sam\lib\threading.py", line 932 in _bootstrap_inner File "E:\Anaconda3\envs\sam\lib\threading.py", line 890 in _bootstrap
Thread 0x0000157c (most recent call first): File "E:\Anaconda3\envs\sam\lib\threading.py", line 306 in wait File "E:\Anaconda3\envs\sam\lib\threading.py", line 558 in wait File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\lib\mailbox.py", line 126 in _wait File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\lib\mailbox.py", line 130 in _get_and_clear File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\lib\mailbox.py", line 283 in wait File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\wandb_run.py", line 224 in _loop_check_status File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\wandb_run.py", line 260 in check_network_status File "E:\Anaconda3\envs\sam\lib\threading.py", line 870 in run File "E:\Anaconda3\envs\sam\lib\threading.py", line 932 in _bootstrap_inner File "E:\Anaconda3\envs\sam\lib\threading.py", line 890 in _bootstrap
Thread 0x00003f8c (most recent call first): File "E:\Anaconda3\envs\sam\lib\site-packages\wandb\sdk\lib\redirect.py", line 640 in write File "E:\Anaconda3\envs\sam\lib\threading.py", line 1202 in invoke_excepthook File "E:\Anaconda3\envs\sam\lib\threading.py", line 934 in _bootstrap_inner File "E:\Anaconda3\envs\sam\lib\threading.py", line 890 in _bootstrap
Current thread 0x00005b04 (most recent call first): <no Python frame>
Hej, it appears you're encountering "Out of Memory" (OOM) errors on your GPU. If the video you're processing is too long or the model/data demands are high, consider these solutions:
-
Reduce the Number of Frames: You can modify the command line argument for the maximum number of frames, for example, by setting
max_frames=10
. -
Use a Lightweight Variant of SAM: Consider replacing the default
ViT-Huge
backbone in SAM with aViT-Tiny
by using MobileSAM or Light HQ-SAM, as instructed here. - Consider GPU Memory Limitations: Your GPU has 2GB of memory, which may be on the lower end for some of the tasks in our repo. I typically run tests with either 12GB or 48GB of GPU memory. If possible, you might find it beneficial to use a GPU with higher memory capacity.
Hope this helps! Please let me know if you have further questions.