首先这是我的环境。
系统信息
操作系统平台和发行版(例如Linux Ubuntu 16.04):Windows10
TensorFlow 安装自(源代码或二进制文件):tensorflow/tensorflow
TensorFlow 版本(使用下面的命令):2.4.0
Python版本:3.7.5
Bazel 版本(如果从源代码编译):3.1.0
GCC/编译器版本(如果从源代码编译):
CUDA/cuDNN 版本:CUDA 11.0.2 / cuDNN8.0.5.39
GPU型号和显存:RTX3090
我编译了tensorflow源并得到'tensorflow.dll'和'tensorflow.lib'。
所以我成功地尝试了下面的代码。
#include <stdio.h>
#include <tensorflow/c/c_api.h>
int main() {
printf("Hello from TensorFlow C library version %s\n", TF_Version());
return 0;
}
接下来,我尝试在 python 中保存 h5 预训练模型(saved_model.pb)后加载它。为此需要一个 C++ 标头,但是当我尝试加载它时,vs2019 中出现链接错误。
#include "tensorflow/cc/saved_model/loader.h"
-问题-
fatal error C1083: cannot open include file. 'Eigen/Core': No such file or directory
所以我手动添加了包含目录的路径。但我收到了这个错误。fatal error C1014: too many include files : depth = 1024
我发现不需要在代码中添加额外的标头。
(我安装了 cuda 和 cudnn 版本,显示在下面链接的底部。)
(https://www.tensorflow.org/install/source_windows?hl=ko)