The radix_sort is already fixed in the nightly release and PyTorch 1.8.1, so you would have to update to one of these versions.
@MCvin your error seems to be different.
Could you rerun the code via CUDA_LAUNCH_BLOCKING=1 python setup.py args and post the complete stack trace (or create a new topic with your error and this information)?