tensorflow / custom-op

Guide for building custom op for TensorFlow
Apache License 2.0
378 stars 115 forks source link

Not finding CUDA header #89

Closed ivs-pychen closed 3 years ago

ivs-pychen commented 3 years ago

When I try to build the gpu op example with the Makefile. I use the command make time_two_op. I got the following error indicated no cuda_fp16.h. image

I have tried the solution like the following picture. But it doesn't work. image