Could you run the code on the CPU and check the error message, please?
This should give you a better error.
If the code runs fine on the CPU, use CUDA_LAUNCH_BLOCKING=1 python setup.py args to rerun the script on the GPU again and post the stack trace here.