CUDA out of memory. Tried to allocate 128.00 MiB. GPU 0 has a total capacty of 21.99 GiB of which 47.00 MiB is free. Process 17346 has 492.00 MiB memory in use. Process 17340 has 492.00 MiB memory in use. Process 17347 has 492.00 MiB memory in use. Process 17363 has 492.00 MiB memory in use. Process 20903 has 3.93 GiB memory in use. Process 6216 has 990.00 MiB memory in use. Process 6219 has 990.00 MiB memory in use. Process 15770 has 3.32 GiB memory in use. Process 26517 has 3.04 GiB memory in use. Including non-PyTorch memory, this process has 2.20 GiB memory in use. Process 7628 has 1.88 GiB memory in use. Process 1036 has 3.64 GiB memory in use. Of the allocated memory 1.75 GiB is allocated by PyTorch, and 150.33 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation.  See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF