RuntimeError: CUDA out of memory. Tried to allocate 1.83 GiB (GPU 0; 15.90 GiB total capacity; 9.39 GiB already allocated; 515.88 MiB free; 5.31 GiB cached)
А можно как-то сделать "куда, го мусор соберём" , или что-то такое?
Я плохо понимаю, как устроено управление памятью ГПУ в питоне, но кажется пришло время разобраться