cuda - 暗网框架无法通过使用 CUDA 的 GPU 加速启动

Question

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

./darknet yolo test cfg/yolo.cfg yolo.weights data/dog.jpg CUDA Error: unknown error darknet: ./src/cuda.c:21: check_error: Assertion `0' failed.

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

谁能帮我解决这个问题？

Answer 1

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