RuntimeError: radix_sort: failed on 1st step: cudaErrorInvalidDevice: invalid device ordinal

问题：错误代码为"cudaErrorInvalidDevice"，表示设备顺序无效。

原因：这可能是由于设备驱动程序或CUDA版本不兼容导致的。可以尝试更新CUDA驱动程序或安装支持的CUDA版本的PyTorch版本。

两张方案解决：

1.这是torch1.8.x的bug，需要将pytorch版本升为1.9.0就可以解决了

2.训练的时候指定显卡进行训练，在训练代码中添加代码

os.environ['CUDA_VISIBLE_DEVICES'] = '1'