Ubuntu лучше для обработки изображений с AI [в ожидании]

Я новичок здесь, я хочу создать свой последний годовой проект по обработке изображений с помощью AI, pytorch, я попробовал его с Windows 10, но он слишком медленный, почти 1 час для одной эпохи! , RAM 6GB, hard: 240 ssd, core i5 intel, если я перехожу с os на ubuntu, то решает ли это, спасибо за ответ