fatal error: third_party/gpus/cuda/include/cuda_fp16.h: 没有那个文件或目录

https://github.com/northeastsquare/bts

打开目录 "tensorflow/include/tensorflow/core/util/gpu_kernel_helper.h"
然后编辑，把

#include "third_party/gpus/cuda/include/cuda_fp16.h" 

替换成

#include "cuda_fp16.h"

然后，再编辑 "tensorflow/include/tensorflow/core/util/gpu_device_functions.h"，把

#include "third_party/gpus/cuda/include/cuComplex.h"
#include "third_party/gpus/cuda/include/cuda.h"

替换成

#include "cuComplex.h"
#include "cuda.h"