(GPU 0; 15.90 GiB total capacity; 9.39 GiB already allocated; 515.88 MiB free; 5.31 GiB cached)
А можно как-то сделать "куда, го мусор соберём" , или что-то такое?
Я плохо понимаю, как устроено управление памятью ГПУ в питоне, но кажется пришло время разобраться
Может просто не надо весь батч в память загружать?
А ты уверен что там мусор?
Обсуждают сегодня