Google Colab训练Yolov9时CUDA设备0请求无效报错求助

问题场景 #

在Google Colab中复现YOLOv9以训练自定义数据集，执行以下训练命令后报错：

!python train_dual.py --workers 8 --batch 4 --img 640 --epochs 50 --data /content/yolov9/data.yaml --weights /content/yolov9-e.pt --device 0 --cfg /content/yolov9/models/detect/yolov9.yaml --hyp /content/yolov9/data/hyps/hyp.scratch-high.yaml

核心报错信息 #

AssertionError: Invalid CUDA '--device 0' requested, use '--device cpu' or pass valid CUDA device(s)

注：日志中出现的cuFFT、cuBLAS注册错误及TensorRT警告为Colab环境中多框架CUDA库冲突导致，不影响训练，可直接忽略。

解决步骤 #

• 启用Colab GPU资源：
  点击顶部菜单栏「代码执行程序」→「更改运行时类型」，在「硬件加速器」下拉选项中选择「GPU」，保存设置。
• 调整训练命令的设备参数：
  • 确认GPU启用后，可保留--device 0参数，或直接省略该参数，让YOLO自动检测可用设备；
  • 若仅需测试代码逻辑，可临时改用CPU：将--device 0替换为--device cpu（训练速度极慢，仅作测试用）。
• 若仍有问题，重启Colab运行时（「代码执行程序」→「重启运行时」）后重新执行训练命令。

内容的提问来源于stack exchange，提问作者Chris Rigano