在 macbook pro / "cuda_runtime.h" 上制作 yolo

Question

我想在我的 MacBook Pro Retina 上使用 YOLOV3（我有一个 NVIDIA GeForce GT 750M）并用我自己的一些图片对其进行训练。

终端中的“make”命令说：include/darknet.h:11:14: fatal error: 'cuda_runtime.h' file not found

所以我下载了所有内容并安装了： - CUDA 驱动版本：418.163（GPU 驱动版本：355.11.10.50.10.103）

这是完整的错误，显示在 mac 终端中：

gcc -Iinclude/ -Isrc/ -DGPU -I/usr/local/cuda/include/ -Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -fPIC -Ofast -DGPU -c ./src/gemm.c -o obj/gemm.o
In file included from ./src/gemm.c:2:
In file included from src/utils.h:5:
include/darknet.h:11:14: fatal error: 'cuda_runtime.h' file not found
    #include "cuda_runtime.h"
             ^~~~~~~~~~~~~~~~

生成 1 个错误。

Answer 1

听错误消息告诉你的内容绝对是一个很好的起点！这似乎是准确的 - 您的文件系统或捆绑目标中缺少 cuda_runtime.h。

如您所见，Cuda docs 提供了确保正确安装驱动程序所需的安装说明。