Выделение памяти с использованием функции cupy создает проблему с выделением нехватки памяти. Потребление памяти из 12 ГБ практически завершено даже до начала тренировки. Теперь во время тренировок вся память расходуется.
Все нормально работает на процессоре.
Я попытался уменьшить размер партии со 100 до одной цифры (4).
Я изменил маленькое задание на numpy, так как Cupy потребляет память.
Я пытался использовать более одного графического процессора, но столкнулся с той же проблемой
please_refer_this
- Обратите внимание, это работает на процессоре *
Результаты хорошие, но мне нужно больше тренироваться, чтобы использовать Cupy и GPU.