Xavier入门教程软件篇-安装TensorFlow-GPU

Xavier入门教程软件篇-安装TensorFlow-GPU

说明：

• 介绍如何为xavier安装TensorFlow-GPU

步骤：

• 安装依赖包：

$ 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

• 安装默认最新的包

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

• 或安装不同的版本

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

• TF_VERSION：The released version of TensorFlow, for example, 1.12.0.
• NV_VERSION：The monthly NVIDIA container version of TensorFlow, for example, 19.01.
• 例如，安装 TensorFlow1.12.0，容器版本 19.01：

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

测试：

• 启动python

$ python3

• 导入

>>> import tensorflow

参考：

• https://devtalk.nvidia.com/default/topic/1042125/official-tensorflow-for-jetson-agx-xavier/
• https://docs.nvidia.com/deeplearning/dgx/install-tf-xavier/index.html