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

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包含您需要的安装说明，以确保正确安装了驱动程序。