在Macbook Pro上制作yolo /“cuda_runtime.h”

Question

我想在我的MacBook Pro Retina上使用YOLOV3（我在那里有NVIDIA GeForce GT 750M，并用一些自己的照片进行训练。

终端中的“ 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具有确保正确安装驱动程序所需的安装说明。