[NEED HELP] Trouble with CUDA capability sm_86

NIT: TORCH_CUDA_ARCH_LIST will be used while building from source and won’t change the shipped compute capabilities in the binaries, which you can get via print(torch.cuda.get_arch_list()).

That’s expected, since Ampere GPUs need CUDA>=11.0

No, the 3070 uses sm_86, which is natively supported in CUDA>=11.1 and is binary compatible to sm_80, so would already work in CUDA=11.0.

In any case, good to hear it’s working now :slight_smile:

2 Likes