imRuntimeError: CUDA error: invalid device ordinal

这个错误通常是由于设备序号不匹配导致的。在使用CUDA时，需要确保设备序号与代码中设置的设备序号一致。如果你遇到了这个错误，可以尝试以下解决方法：
1. 检查代码中设置的设备序号是否正确，确保与实际设备序号一致。
2. 尝试将设备序号改为0或其他可用的序号。
3. 在代码中添加以下语句：`%env CUDA_LAUNCH_BLOCKING=1`，以确保CUDA kernel错误能够同步报告。
如果以上方法都无法解决问题，你可以尝试查看更详细的错误信息，以便更好地定位问题所在。