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()).



 Ilias_Giannakopoulos:

also tried with 10.2, but didn’t have any luck


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



 Ilias_Giannakopoulos:

Could it be that NVIDIA 3070 works with CUDA 11.2 and higher, while PyTorch supports up to CUDA 11.1 for the moment?


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