You might need to disable the NCCL build as the min. compute capability seems to be 3.5 based on this definition 1. Once this is done, you would then have to check if the current PyTorch source has the same min. compute capability.