Is it possible to build pytorch from source for arm with nvidia gpu

1

Is it possible to build pytorch from source for arm with nvidia gpu?

2

Yes, that’s possible and we are already building PyTorch for ARM e.g. via these NGC containers.

3

Thank you! I’ll try it out. But I’m wondering why there is no official python lib for arm gpu.

4

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.

5

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!