Does pytorch support building from source on arrch64

torch.compile
1

Does pytorch support building from source on arrch64?
when I build pytorch2.1.2 cu118 with python3.9, get error about"avx".but I saw others build gpu pytorch v2.0.1 version succeed https://blog.csdn.net/hknaruto/article/details/132692665

2

Yes, PyTorch supports ARM and you can download pre-built binaries for Jetson devices, ARM servers with NVIDIA GPUs, newer Macs, and CPU-only ARM builds.

3

thanks，I did it，but I have another question:I am compiling the torch gpu version on the ARM architecture, but I find that the computing power of gpu and glibc cannot be reused. For example, the whl package of torch compiled on the computing power sm70 cannot be used on other computing gpu. The whl package compiled on glibc2.3.30 cannot be used on operating systems earlier than this glibc version. Is there a way to compile a relatively general torch?

4

Yes, add the desired GPU architecture via TORCH_CUDA_ARCH_LIST and use e.g. a manylinux environment with a low glibc to build the wheels.