Cupy подавляет игнорируемые исключения OutOfMemory ⇐ Python

[Anonymous]

При использовании Cupy я получаю массу сообщений об игнорируемых ошибках OutOfMemoryError, например: как показано ниже:

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

Exception ignored in: 'cupy.cuda.thrust.cupy_malloc'
Traceback (most recent call last):
File "cupy/cuda/memory.pyx", line 738, in cupy.cuda.memory.alloc
File "cupy/cuda/memory.pyx", line 1424, in cupy.cuda.memory.MemoryPool.malloc
File "cupy/cuda/memory.pyx", line 1445, in cupy.cuda.memory.MemoryPool.malloc
File "cupy/cuda/memory.pyx", line 1116, in cupy.cuda.memory.SingleDeviceMemoryPool.malloc
File "cupy/cuda/memory.pyx", line 1137, in cupy.cuda.memory.SingleDeviceMemoryPool._malloc
File "cupy/cuda/memory.pyx", line 1382, in cupy.cuda.memory.SingleDeviceMemoryPool._try_malloc
File "cupy/cuda/memory.pyx", line 1385, in cupy.cuda.memory.SingleDeviceMemoryPool._try_malloc
cupy.cuda.memory.OutOfMemoryError: Out of memory allocating 5,607,143,424 bytes (allocated so far: 7,307,858,432 bytes).

Насколько я понимаю, эти сообщения генерируются Cupy, который пытается выделить некоторую память, но терпит неудачу, поскольку память графического процессора заполнена выделенными, но неиспользуемыми (т. е. областями кэша) блоками. После неудачного выделения Cupy освобождает некоторые выделенные, но неиспользуемые блоки и повторяет попытку выделения, и все проходит успешно. Поскольку сгенерированное сообщение об ошибке связано с управлением внутренней памятью Cupy, а приложение работает нормально, отображение этих сообщений на консоли очень раздражает.
Есть ли какой-нибудь способ подавить такие сообщения? Эти сообщения делают консоль совершенно беспорядочной и непригодной для использования.

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