Регрессия GPyTorch с KeOps — OutOfMemoryError ⇐ Python

[Anonymous]

Я пытаюсь подогнать к моему набору данных некоторую большую (относительно размера памяти графического процессора) модель гауссовского процесса, используя GPyTorch с KeOPS. Однако, похоже, я не могу извлечь выгоду из KeOPS для уменьшения объема памяти.
Я начал с руководства по GPyTorch с KeOPS, взятого с GitHub здесь. При запуске ноутбука появляется следующее сообщение об ошибке: OutOfMemoryError: CUDA не хватает памяти. Пытался выделить 528,38 ГиБ.. Я не думаю, что ноутбук ожидает, что у пользователя будет более 500 ГБ оперативной памяти графического процессора.
Что я делаю неправильно? Или пример устарел? Сейчас используется GPyTorch 1.13.

Подробнее здесь: https://stackoverflow.com/questions/792 ... emoryerror