Even I am getting the same error when I run python setup.py install
nvcc --version output :
nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2015 NVIDIA Corporation Built on Tue_Aug_11_14:27:32_CDT_2015 Cuda compilation tools, release 7.5, V7.5.17
PS : NO_DISTRIBUTED=1 python setup.py install is working but my question is what is NO_DISTRIBUTED=1 for?
Edit 1: As I previously mentioned, I have installed it via NO_DISTRIBUTED=1 python setup.py install but am unable to user affine_grid function.