New to this forum,
I recently built Pytorch 1.0 on the Jetson Xavier. I had to modify several CMake files to get it to link Openblas and build/link Aten correctly for aarch64.
aten/CMakeLists.txt
cmake/Modules/FindBLAS.cmake
cmake/public/cuda.cmake --> changed to cuda 10
Initially I followed the Jetson Hacks instructions but it seems that by default pytorch was linked to a lapack (generic) library that was not cuda enabled. /usr/lib/aarch64-linux-gnu was not on the search path?
Perhaps I missed a configuration flag when targeting AARCH64? I would like to submit these changes (maybe on a branch) if they are not already there.
Thanks,
Rich