我的操作系统是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++了解的不是很多,希望大神可以帮我解决这个问题。
YOLO在python中调用设置darknet.set_gpu(1)无效
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
2条回答 默认 最新
夜夜0810 2018-12-11 07:12关注我也出现了同样的问题,最后发现是libdarknet.so文件的问题。你可能是用了cpu编译生成的libdarknet.so文件。
评论 打赏 举报解决 1无用