Thank you for sharing this great project. I would lie to know what is the required pytorch and cuda version?