Does Pytorch support ARM processor(aarch64)?

smth · April 11, 2017, 6:43pm

it does support aarch64 processor. Someone from NVIDIA wrote how to build it on their jetson platform:

gist.github.com

https://gist.github.com/dusty-nv/ef2b372301c00c0a9d3203e42fd83426

pytorch_jetson_install.sh

#!/bin/bash
#
# pyTorch install script for NVIDIA Jetson TX1/TX2,
# from a fresh flashing of JetPack 2.3.1 / JetPack 3.0 / JetPack 3.1
#
# for the full source, see jetson-reinforcement repo:
#   https://github.com/dusty-nv/jetson-reinforcement/blob/master/CMakePreBuild.sh
#
# note:  pyTorch documentation calls for use of Anaconda,
#        however Anaconda isn't available for aarch64.

This file has been truncated. show original