在CPU上训练YOLOv7会产生CUDA错误

Question

我正在尝试在没有gpu的情况下运行train一个yolov7模型。这是我目前在colab上使用的命令行。

python train_aux.py --workers 1 --device cpu --batch-size 1 --data data/coco.yaml --img 128 128 --cfg /content/yolov7/cfg/training/yolov7-e6e.yaml --weights '' --name yolov7-e6e --hypdata/hyp.scratch.p6.yaml`

出于某种原因，我第一次收到警告

warnings.warn('User provided device_type of 'cuda', but CUDA is not available. Disabling')

然后我得到错误

RuntimeError: No CUDA GPUs are available

在第一个时期。我不明白当我在CPU上运行它时，它为什么要使用cuda。我是不是错过了一些必须在代码中编辑才能修复的地方？这是我正在使用的github的链接

我试着下载了有助于使用的cuda库实例。

!pip install cuda-python

但它并没有解决问题。

Answer 1

因此，这个问题似乎是由于某些过程的cuda被硬编码到模型中。更深入的解释可以在这里找到链接。同时，由于某种原因删除--device cpu修复了它。