nvidia jetson agx xavier搭建tensorflow-gpu环境

?????????????????
nvidia jetson agx xavier这个小家伙目前释放出最新的jetpack 4.2.1，也是我的刷机版本，刷机之后自带cuda10，所以我的xavier环境：

Ubuntu 18.04 LTS aarch64
jetpack 4.2.1
python 3.6
cuda 10

官方给出的组分：

see？一样滴。

搭建tf-gpu，直接：

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu --user

相关依赖包会被自动安装。但如果在依赖包安装过程中h5py报错（即我遇到的问题），则：

sudo apt-get install libhdf5-serial-dev hdf5-tools

或者可以先装载依赖包：

sudo apt-get install libhdf5-serial-dev hdf5-tools
sudo apt-get install python3-pip
pip3 install -U pip
sudo apt-get install zlib1g-dev zip libjpeg8-dev libhdf5-dev 
sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker grpcio six mock requests gast h5py astor termcolor

然后再执行安装：

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu --user

则不会有h5py报错的问题。

若jetpack版本为4.1.1，则：

pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v411 tensorflow-gpu==1.13.0rc0+nv19.2 --user

以上获取的都是最新的tf-gpu支持，若想获取指定版本，可以：

sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==$TF_VERSION+nv$NV_VERSION

其中：
TF_VERSION：tensorflow版本，比如1.12.0。
NV_VERSION：nvidia每月的tensorflow容器版本，比如19.01。
完整命令类似：

sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.12.0+nv19.1

hope u enjoy?????????