Hi,
Was there anything else printed before the error is raised?
Also you can run your code with CUDA_LAUNCH_BLOCKING=1 otherwise the python stack trace is not correct (as cuda calls are asynchronous).
Also if you’re in an interpreter. After one such assert is raised, you need to restart the interpreter as it puts the GPU in a bad state and any attempt to use the GPU will through an error.