I’m looking for a way to compile (install) pytorch for arm64 platform and as of now I’m not interesting of optimized (GPU-like) version of pytorch, it’s good enough to have CPU version able to run on ARM device. Please can you advice how can I either compile or install such a version of pytorch?
In my case I do not have CUDA available on device but instead I have OpenCL 1.2 FP.
I think you can just compile with NO_CUDA=1
. Could you try it and see if there is any error?
I would love too, but is there install dependencies ? Since my platfrom ti’s kind of minimal on packages …
Yes, you would need numpy pyyaml mkl setuptools cmake cffi
Is it mkl library for arm architecture? Isn’t Intel lib?
Right, you can skip that. Sorry I just copied + pasted from Github install page without checking.
I’m wondering to know if there’s a way to cross compile Pytorch on Allwinner platform (AArch64 processor). Leads are greatly appreciated.
Also @Peter_Peter, I presume that you’ve compiled or installed pytorch on arm64 by now. Could you please post the steps you have followed for the same?
TIA
@NikhilKanda this is the script I used to compile on AGX Xavier which is an ARM64 platform: https://gist.github.com/soumith/45f61db3379db67414814ad9d5f2a427
oh, sorry yes this is not cross-compilation.
I dont think all parts of our build system are even ready for cross-compilation
Can you give me any information about cross compilation of pytorch …
I haven’t tried this, but the following tutorial looks relevant: