在 CPU 上训练 YOLOv7 会出现 CUDA 错误答案

【问题标题】：training YOLOv7 on CPU provides CUDA error在 CPU 上训练 YOLOv7 会出现 CUDA 错误
【发布时间】：2022-12-18 08:41:04
【问题描述】：

我正在尝试在没有 gpu 的情况下训练 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 的 link

我试图下载有助于使用的 cuda 库。

!pip install cuda-python

但它没有解决问题。

【问题讨论】：

尝试删除 --device cpu 参数。在没有 GPU 的情况下运行时，这对我有用。
@nithins 是的，这似乎出于某种原因有效，奇怪的是 cpu 不起作用

标签： python google-colaboratory yolo

【解决方案1】：

所以看起来这个问题是由于 cuda 被硬编码到某些程序的模型中。可以在此处找到更深入的解释link。同时由于某种原因删除--device cpu修复了它。

【讨论】：