I've been trying to fix this for over a week. Running the official template with pytorch 2.1.0, cuda 118