RuntimeError: CUDA error: invalid device ordinal

在运行相关的脚本时，报错：

 我遇到的问题是，GPU数量不够，只有一个。

解决方法：首先在终端运行：

nvidia-smi

 查看此设备有多少GPU可用（我的是只有一个）

 然后将对应文件中的gpu数量改为0（gpu都是从0开始编号的）

同时，检查在代码中有其他地方用了GPU设备，如将：

device = torch.device("cuda:1")  # 这应该改为 cuda:0

在IDE中可用高级检索功能 ，通过 Ctrl + Shift + F（Windows/Linux）:

若是这种，就不需要修改。