Jetson Nano & TX2 （六）Tensorflow

一、基础依赖

系统：

$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran

更新pip3:

$ sudo apt-get install python3-pip
$ sudo pip3 install -U pip testresources setuptools

 

二、选择特定版本

JP_VERSION ：

                         42 对于JetPack 4.2.2 或 33 对于JetPack 3.3.1

安装 特定版本 的JetPack支持的TensorFlow的最新版本，请发出以下命令：

$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION tensorflow

TF_VERSION：

                         TensorFlow的发行版本

NV_VERSION：

                         NVIDIA容器版本

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

 

ps：

    小技巧 网页查找：https://developer.download.nvidia.cn/compute/redist/jp/$JP_VERSION 所包含的wheel

三、例子

• JetPack 4.2.2
• TensorFlow 1.13.1
• NV 19.03

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

 

四、其他

                         官方文档：地址