Pytorch, CUDA, memory Подскажите пожалуйста, как править ошибку с памятью? Везде пишут,

Question

Big Data Science :: AI / Big Data / Machine Learning / MLOps

Pytorch, CUDA, memory Подскажите пожалуйста, как править ошибку с памятью? Везде пишут,

что помогает уменьшать размер батча, но у меня уже и так 1. Где память-то?
RuntimeError: CUDA out of memory. Tried to allocate 54.00 MiB (GPU 0; 11.77 GiB total capacity; 8.90 GiB already allocated; 62.62 MiB free; 9.03 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

Если включить пайтоновский интерпретатор в этом виртуальном окружении и импортировать pytorch, то вот что получаю:
>>> torch.cuda.memory_allocated()
0
>>> torch.cuda.memory_reserved()
0
>>> torch.cuda.max_memory_reserved()
0
>>> torch.cuda.max_memory_allocated()
0