暗网框架无法启动使用 CUDA 的 GPU 加速答案

【问题标题】：Darknet framework fails to start with GPU acceleration using CUDA暗网框架无法启动使用 CUDA 的 GPU 加速
【发布时间】：2016-10-17 11:50:08
【问题描述】：

我试图通过使用 CUDA API 的 GPU 加速来运行 Darknet。所以我按照这里的说明，在 Makefile 中更改了 GPU=1 并开始制作。当我尝试运行测试时，由于 CUDA 错误而失败。

./darknet yolo test cfg/yolo.cfg yolo.weights data/dog.jpg CUDA 错误：未知错误暗网：./src/cuda.c:21: check_error: Assertion `0' 失败了。

我在 Titan X 上使用 Ubuntu 14.04、CUDA 7.5 和我的 NVIDIA-SMI 352.93 和驱动程序版本：352.93 我很确定我的 CUDA 工作正常并且驱动程序的版本是最新的，因为我正在使用它来加速 Caffe。我的猜测是 Darknet 找不到 CUDA 目录。

谁能帮我解决这个问题？

【问题讨论】：

你的代码是基于这个repo吗？

标签： cuda neural-network deep-learning

【解决方案1】：

您可以通过阅读第 21 行 ./src/cuda.c 中的代码并检查哪个断言失败（如错误消息所示）来找出原因。

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

【讨论】：

【解决方案2】：

您应该修改配置参数“ARCH”的值。默认值为--gpu-architecture=compute_52，--gpu-code=compute_52。我的设置是--gpu-architecture=compute_30，--gpu-code=compute_30，它可以工作。这取决于您的实际 gpu 架构。更多细节在 cuda 工具包文档中。

【讨论】：

OP 表示他们有一个 Titan X GPU，而对于 Titan X GPU，实际的计算架构是 compute_52，所以应该没有必要改变它。我认为这个建议没有任何意义。
您的答案适用于 Tesla K80 GPU。感谢您发布它，在其他任何地方都找不到此信息。

【解决方案3】：

同意“user6568204”。另外，你可以修改\Makefile中配置参数'ARCH'的值，你可以在这个网站找到其他'ARCH'值： http://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#virtual-architecture-feature-list 或者看这张图virtual-architecture-feature-list

【讨论】：

【解决方案4】：

也许你应该修改 Makefile 中的 ARCH。你的 GPU 分数可以在这个站点找到：https://developer.nvidia.com/cuda-gpus

【讨论】：

【解决方案5】：

我刚刚遇到这个问题并解决了。我的解决方案是

sudo rm -rf ~/.nv

然后重启。

【讨论】：

【解决方案6】：

我遇到了同样的问题。修改 Makefile 中的 ARCH 解决了我的问题。我的 GPU 是 GTX 1080（我使用 nvidia-smi 进行了检查）

> $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] \

【讨论】：

我的 cuda 工具包版本是 cuda 9.1，我的 linux 版本是 ubuntu 16.04