如何在cpu模式下使用pycaffe进行预测?

时间:2019-10-17 12:15:17

标签: caffe pycaffe

我试图用python代码进行预测,例如:

caffe.set_mode_cpu()
net = caffe.Net(proto, mod, caffe.TEST)

但是发生错误

Check failed: error == cudaSuccess (38 vs. 0)  no CUDA-capable device is detected

如何处理?

0 个答案:

没有答案