устранение ошибок

[astro_ivanovich]

ругается на cuda memory allocator – что делать?

[Dmitronik]

обычно помогает понизить batch size до 1 или снизить разрешение. если не помогает, пробуй обновить драйвер nvidia до версии 531 или перейти на 532, иногда помогает