Closed MEIXuYan closed 5 years ago
See the updated build instructions in the README.md from commit 129cfefebf1dffa074d030be94065bdebc74f830. The copy-and-paste solution would be
cd /tmp
wget https://github.com/NVlabs/cub/archive/v1.8.0.zip
unzip v1.8.0.zip -d $HOME/libs
export CUB_INC=$HOME/libs/cub-1.8.0/
rm /tmp/v1.8.0.zip
Feel free to ask, if there is anything unclear.
Thanks for sharing your code. I got a problem when complie the usr_op, showing as below: (I'm not familiar with the Cuda-CUB)
I'm work with ubuntu 14.04, cuda 9.0, tensorflow-gpu 1.9.0, GTX 1080Ti. Hope you can fix it, thank you.