Darknet框架无法使用CUDA启动GPU加速

我试着用CUDAAPI用GPU加速运行Darknet。所以我按照这里的说明，在Makefile中更改了GPU=1并开始make。 当我试图运行测试时，由于CUDA错误，它失败了

./darknet yolo test cfg/yolo.cfg yolo.weights data/dog.jpg CUDA错误： 未知错误darknet:./src/cuda.c:21:检查错误：断言“0” 失败了

我在泰坦X上使用的是Ubuntu 14.04、CUDA 7.5和我的NVIDIA-SMI 352.93以及驱动程序版本：352.93 我很确定我的CUDA运行良好，驱动程序版本是最新的，因为我用它来加速Caffe。 我猜Darknet找不到CUDA目录

---

有人能帮我解决这个问题吗？

您可以通过阅读第21行

/src/cuda.c

中的代码，并检查哪个断言失败，如错误消息所示

./src/cuda.c:21: check_error: Assertion `0' failed.

---

您应该修改配置参数“ARCH”的值。默认值为--gpu架构=计算单元52，-gpu代码=计算单元52。我的设置是--gpu架构=compute_30，-gpu代码=compute_30，它可以工作。这取决于您的实际gpu架构。更多详细信息请参见cuda工具包文档

同意“用户6568204”。 此外，您可以在\Makefile中修改配置参数“ARCH”的值，并且可以在此网站中找到其他“ARCH”值：

---

或者看看这张图片

也许你应该修改Makefile中的ARCH。你的GPU分数可以在这个网站上找到：

我刚刚遇到这个问题并解决了它。我的解决办法是

sudo rm-rf~/.nv

然后重新启动。

我也有同样的问题。 在Makefile中修改ARCH解决了我的问题。 我的gpu是GTX 1080（我使用nvidia smi进行了检查）

我发现这里的设置适合我的gpu

http://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/

我在Makefile中更改了arch设置，如下所示

ARCH= -gencode arch=compute_61,code=[sm61,compute_61] \

---

您的代码是基于此的吗？OP表示他们有一个Titan X GPU，对于Titan X GPU，实际的计算架构是compute_52，因此不需要更改它。我认为这个建议没有任何意义。你的答案适用于特斯拉K80 GPU。感谢您的发布，在其他地方找不到此信息。我的cuda工具包版本是cuda 9.1，linux版本是ubuntu 16.04

ARCH= -gencode arch=compute_61,code=[sm61,compute_61] \