Как освободить место на GPU?
Я обрабатываю свою модель на очень большом наборе данных (размером около 1 ТБ), разбивая его на куски, а затем запускаю прогноз для каждого куска и затем добавляю результат в файл.
После запуска каждого блока я должен освободить память графического процессора, чтобы можно было обрабатывать и прогнозировать другой блок.
Вместо того, чтобы снова и снова запускать "Перезапуск ядра", я могу запустить цикл, если существует какая-либо команда для очистки памяти графического процессора.