AssertionError: Invalid device id

这个错误通常是因为你在使用 CUDA 的时候，指定了一个不存在的设备 ID。可以尝试以下几种解决方法：

1. 检查一下你的 GPU 是否正常，以及是否已经正确安装了 CUDA。

2. 使用 `torch.cuda.device_count()` 函数来查看你的系统中有多少个可用的 GPU 设备。

3. 通过设置环境变量 `CUDA_VISIBLE_DEVICES` 来限制你使用的设备 ID。例如，如果你想使用第 0 号设备，可以运行以下命令：

   export CUDA_VISIBLE_DEVICES=0

这个命令会将你的环境变量设置为只有第 0 号设备可见，从而避免了设备 ID 不存在的问题。

希望这些方法能够帮到你。