RuntimeError: CUDA error: invalid device ordinal

这个错误通常是因为你在尝试使用一个不存在的 CUDA 设备。你可以使用以下代码来检查可用的 CUDA 设备：

import torch
if torch.cuda.is_available():
    device_count = torch.cuda.device_count()
    print(f"Found {device_count} available device(s).")
    for i in range(device_count):
        print(f"Device {i}: {torch.cuda.get_device_name(i)}")
else:
    print("CUDA is not available.")

运行这段代码后，你应该能够看到所有可用的 CUDA 设备以及它们的编号。如果你仍然遇到问题，请检查你的环境变量和 CUDA 版本是否正确配置。