CUDA and torch version used in this project looks outdated. Can I use CUDA 12.1 instead or does this cause bug?