Anyone with ML development experience know what changes are needed to make this work on a CPU without a CUDA GPU? Seems heavily coupled to CUDA.
Maxed out at 4 GB RAM for 256x144 images for me.
PyTorch CPU installation instructions: https://pytorch.org/get-started/previous-versions/
OpenCV should work without CUDA. If not, build from source and consider `WITH_CUDA` flag.