Crosscompilation for arm64

(Peter Peter) #1

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.

(Simon Wang) #2

I think you can just compile with NO_CUDA=1. Could you try it and see if there is any error?

(Peter Peter) #3

I would love too, but is there install dependencies ? Since my platfrom ti’s kind of minimal on packages …

(Simon Wang) #4

Yes, you would need numpy pyyaml mkl setuptools cmake cffi

(Peter Peter) #5

Is it mkl library for arm architecture? Isn’t Intel lib?

(Simon Wang) #6

Right, you can skip that. Sorry I just copied + pasted from Github install page without checking. :slight_smile: