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

我的操作系统是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个回答

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

显卡为GT750M???我的显卡也是GT750M,测试时可以使用gpu,但是训练时感觉没有使用gpu,,,,一个batch 的75s 太慢了 老哥,你最好怎么解决的 我联系微信17691070209

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问
相关内容推荐