RuntimeError: CUDA error: invalid device ordinal

[OneStepAndTwoSteps]

作者你好，我在colab上运行项目的时候出现了这样的问题，看代码中的cuda指定的是0也没有指定多个GPU。我不知道是什么原因导致的这个问题。
RuntimeError: CUDA error: invalid device ordinal
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be
incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

[zbisure]

demo.py文件的
AutoModel.from_pretrained(gen_model_name_or_path, trust_remote_code=True).half().cuda(1)
cuda(1)改为cuda(0)