Oh okay that makes sense. The machine I’m on currently is Windows so I’m setting CUDA_LAUNCH_BLOCKING=1 in the system settings.
I’ll try that, thank you. I’m also logging GPU memory to see if OOM is the problem