Я пытаюсь запустить код pytorch с помощью ноутбука jupyter и получил эту ошибку
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.23 GiB already allocated; 18.83 MiB free; 1.25 GiB reserved in total by PyTorch)
Я уже нашел ответ. и, прежде всего, просто уменьшите размер партии. Я попытался уменьшить размер пакета с 20 до 10 до 2 и 1. Прямо сейчас все еще не могу запустить код
Мой графический процессор - Quadro p620