#pytorch
Вопрос:
Почему он занимает так много памяти GPU, когда я его использую .cuda() ? И как его уменьшить? Моя бедная память графического процессора 1660-х годов настолько ограничена.
Комментарии:
1. Вы убедились, что он был пуст, когда запускали свой процесс python?
Ответ №1:
Выполнять—
torch.cuda.empty_cache()
Это уменьшит объем памяти графического процессора. Если это не уменьшит объем памяти, то перезапуск ядра решит проблему.
Комментарии:
1. это не работает. и после перезагрузки ядра, когда я использую .cuda(), он по-прежнему занимает много памяти GPU, как я писал выше.