Jetson Nano 安装 TensorFlow

前言

Jetson Nano 的处理器架构为 ARM,pypi 镜像并未提供 ARM

版 TensorFlow 的轮子,因此无法使用 pip3 install tensorflow 进行安装。

但是 NVIDIA 自己提供了轮子,可以下载进行安装。

操作记录

准备工作

sudo apt update && sudo apt install -y python3-pip

pip3 install -i https://mirrors.bfsu.edu.cn/pypi/web/simple pip -U

pip3 config set global.index-url https://mirrors.bfsu.edu.cn/pypi/web/simple
sudo apt install -y libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran cython3
pip3 install wrapt six grpcio protobuf numpy gast keras-preprocessing termcolor werkzeug markdown wheel absl-py tensorboard google-pasta astor   future mock futures tensorflow-estimator h5py keras-applications

打开 Jetson 下载中心: https://developer.nvidia.com/zh-cn/embedded/downloads

然后搜索 tensorflow

可以选择目前最新的版本进行下载

然后使用 pip3 install [轮子文件路径] 进行安装,

比如这里对应的 pip3 install tensorflow-2.5.0+nv21.8-cp36-cp36m-linux_aarch64.whl