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

 

四、其他

                         官方文檔：地址