我有Coral AI usb TPU,我成功地运行了入门示例,部署了已经编译/训练过的示例模型(图像分类),并在TPU上运行了一个鹦鹉图像推理:--input test_data/parrot.jpg
但是,我想在计算机CPU (例如,我的笔记本电脑或Raspberry Pi )中运行相同模型的推理,以比较在加速器中运行推理所需的时间,比如Coral AI和通用CPU。我所能找到的只有以下文件:,特别是
我有一个C++程序,使用Tensorflow 2运行卷积神经网络的推理。该程序运行在具有专用GPU的服务器上,所期望的行为是在GPU上运行的推断。在GPU出现故障时,Tensorflow开始使用CPU而不是GPU。有没有使用Tensorflow C++ API来检查Tensorflow是否正在使用CPU的方法?在C++失败的情况下,是否有任何方法可以避免Tensorflow切换到CPU?