>>937309809
You just need to update pytorch to a version that supports cuda 12.8, at least from a quick search.