Before command "python setup.py install" you better command export CUDA_HOME="/usr/local/cuda/" , this is your cuda path.