Как я могу опустошить память графического процессора, когда получаю ошибку, как OOM?Python

Программы на Python
Ответить Пред. темаСлед. тема
Anonymous
 Как я могу опустошить память графического процессора, когда получаю ошибку, как OOM?

Сообщение Anonymous »

Иногда, когда я получил ошибку OOM, но параметры LLM загружались в GPU, и ее нельзя очистить автоматически. < /p>
Итак, я пробую это < /p>

Код: Выделить всё

torch.cuda.empty_cache()
Но это не работало.>

Подробнее здесь: https://stackoverflow.com/questions/797 ... r-like-oom
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

Вернуться в «Python»