#python-3.x #memory-management #pytorch
Вопрос:
Я пытаюсь запустить GAN над некоторыми изображениями размером 256×256. Однако всякий раз, когда я начинаю тренироваться, я получаю эту ошибку:
RuntimeError: CUDA out of memory. Tried to allocate 2.19 GiB (GPU 0; 10.00 GiB total capacity; 437.15 MiB already allocated; 6.38 GiB free; 1.51 GiB reserved in total by PyTorch)
Я не уверен, как это решить. Кажется, что памяти много, поэтому я не понимаю, почему это произошло.
Комментарии:
1. Попробуйте меньшие размеры партий.