如何控制 GPU 委托回退（C API）答案

【问题标题】：How to control GPU delegate fallback (C API)如何控制 GPU 委托回退（C API）
【发布时间】：2020-09-16 09:14:08
【问题描述】：

我正在编写一个依赖于 tensorflow lite GPU v2 功能link 的程序。在初始化步骤，我正在尝试创建 GPU v2 委托。但是，在一些设备上，所需的 OpenCL 库不可用，我想使用 CPU 委托而不是自动 GPU v1 (GLES) 回退。

我尝试了以下代码，但即使 OpenCL 不可用，也会创建委托（非空）。

TfLiteDelegate* delegate = TfLiteGpuDelegateV2Create(&optionsV2);

if (delegate == nullptr) LOGV("launchDelegate", "==> Failed to create GPU v2 delegate");
TfLiteInterpreterOptionsAddDelegate(options, delegate);

如何检测到 GPU v2 委托在初始化时不可用，以便我可以控制回退选项？

【问题讨论】：

标签： tensorflow tensorflow2.0 tensorflow-lite

【解决方案1】：

您能分享一下您使用optionsV2 指定的选项吗？启用TFLITE_GPU_EXPERIMENTAL_FLAGS_CL_ONLY 实验标志可能值得一试。

https://github.com/tensorflow/tensorflow/blob/5874c1424db542293276cdaeb21f8de9febabd60/tensorflow/lite/delegates/gpu/delegate.h#L56

【讨论】：

【解决方案2】：

或者您可以自行检查 libOpenCL.so 的可用性。

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/delegates/gpu/cl/opencl_wrapper.cc#L81

【讨论】：