Askfk 2018-04-11 12:53 采纳率: 0%
浏览 4054
已结题

YOLO在python中调用设置darknet.set_gpu(1)无效

我的操作系统是macos,显卡为GT750M,已经成功安装了cuda,在终端调用yolo官网的示例代码可以实现gpu运算,但在python中按照作者封装好的darknet.py调用yolo时设置darknet.set_gpu(1)却无法使用gpu运算,仍是cpu运算。
我在/src/cuda.c中把cuda_set_device函数的gpu_index直接设置为1,仍无法使用gpu运算,我对c++了解的不是很多,希望大神可以帮我解决这个问题。

  • 写回答

2条回答 默认 最新

  • 夜夜0810 2018-12-11 07:12
    关注

    我也出现了同样的问题,最后发现是libdarknet.so文件的问题。你可能是用了cpu编译生成的libdarknet.so文件。

    评论

报告相同问题?