1.
#include "tensorflow/core/util/cuda_kernel_helper.h"
改为
#include "tensorflow/core/util/gpu_kernel_helper.h"
2.
#include "third_party/gpus/cuda/include/cuda_fp16.h"
改为
#include "cuda_fp16.h"
3.
#include "third_party/gpus/cuda/include/cuComplex.h"
改为
#include "cuComplex.h"
4.
#include "third_party/gpus/cuda/include/cuda.h"
改为
#include "cuda.h"
5.
/usr/bin/ld: cannot find -ltensorflow_framework
解决方法:
在目录 ~/anaconda3/envs/XXX/lib/python3.6/site-packages/tensorflow 下,将文件libtensorflow_framework.so.1 (或者有libtensorflow_framework.so.2) 复制并粘贴为 libtensorflow_framework.so
即:cp libtensorflow_framework.so.1 libtensorflow_framework.so