Is it possible to build pytorch from source for arm with nvidia gpu?
Yes, that’s possible and we are already building PyTorch for ARM e.g. via these NGC containers.
Thank you! I’ll try it out. But I’m wondering why there is no official python lib for arm gpu.
I’m not sure what “Python lib for ARM GPU” means, since the linked container is compiled for an ARM CPU and works on any GPU with a compute capability >= 5.2.
I’m sorry I’m new to python. I tried to install torch with ‘pip install torch’ in arm with gpu, but torch.cuda.is_available() is false. Anyway, NGC containers work for me. Thanks!