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

Question

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

终端中的 "make"-命令显示：include/darknet.h:11:14: 致命错误：'cuda_runtime.h' 找不到文件

所以我下载了一切，安装： - 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 包含确保驱动程序安装正确所需的安装说明。

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

make yolo on macbook pro / "cuda_runtime.h"

terminal

makefile

yolo