the Caffe::mode_ variable that controls this is thread-local, so ensure you’re calling caffe.set_mode_gpu() in each thread before running any Caffe functions. That should solve your issue.
Caffe set_mode GPU 在多线程下失效
最新推荐文章于 2018-08-05 15:57:47 发布
the Caffe::mode_ variable that controls this is thread-local, so ensure you’re calling caffe.set_mode_gpu() in each thread before running any Caffe functions. That should solve your issue.