сука какой же pytorch хуйня

A = <data>

использует 300MiB

A = A @ A.T

использует 12GB

ну ок, но я же присвоил, наверное он освободит память как только понадобится

НИХУЯ, CUDA OUT OF MEMORY

ну окей, нагуглили функцию для принудительного освобождения памяти

НИХУЯ, CUDA OUT OF MEMORY

сука у вас GC в языке для кого, или он только по CPU OOM начинает освобождать