Ошибка памяти Cuda pytorch на свежем экземпляре p2.2xlarge?

#amazon-ec2 #pytorch

Вопрос:

Я получаю следующую ошибку на совершенно новом экземпляре p2.2xlarge ec2? Почему pytorch занимает так много памяти — это именно то, что он потребляет до сих пор, и что вызвало ошибку памяти?

 RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 11.17 GiB total capacity; 10.72 GiB already allocated; 71.88 MiB free; 10.79 GiB reserved in total by PyTorch  

Комментарии:

1. Вы предоставляете очень мало информации, у меня были аналогичные проблемы, в самом простом случае (хотя есть и другие довольно сложные проблемы) Я бы взглянул на размер партии. Увеличение или уменьшение размера пакета может быть эффективным (если вы предоставите более подробную информацию, я мог бы помочь вам в дальнейшем).