Build "slice" error

[BaiYu0120]

@mgharbi, I got a error when build custom Tensorflow operator to "slice" in the bilateral grid, log as follows. Can you help me to fix it? I got TF 1.4 and Ubuntu 16.04, Cuda 8.0 in my PC. //////////////////////////////// nvcc -std c++11 -c ops/bilateral_slice.cu.cc -o build/bilateral_slice.cu.o -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -Ipython -c 'import tensorflow as tf; print(tf.sysconfig.get_include())' -Iops -gencode=arch=compute_30,code="sm_30,compute_30" -expt-relaxed-constexpr -Wno-deprecated-gpu-targets -ftz=true --ptxas-options=-v -lineinfo In file included from ops/bilateral_slice.cu.cc:20:0: /usr/local/lib/python2.7/dist-packages/tensorflow/include/tensorflow/core/util/cuda_kernel_helper.h:24:31: fatal error: cuda/include/cuda.h: No such file or directory compilation terminated. Makefile:32: recipe for target 'build/bilateral_slice.cu.o' failed make: *** [build/bilateral_slice.cu.o] Error 1 /////////////////////////////////

[cchen156]

remove this: -DGOOGLE_CUDA=1

[butterl]

add ../$(CUDA_HOME) to -I make it

[TigerStone93]

Refer this https://devtalk.nvidia.com/default/topic/1027553/cuda-setup-and-installation/importerror-libcublas-so-8-0-cannot-open-shared-object-file-no-such-file-or-directory/