Jetson Nano Tensorflow-GPU 安裝

1. Installing dependencies

在安裝 tensorflow-gpu 前，你需先安裝以下套件:

a. NVIDIA SDK Manager

b. HDF5: 是一種為存儲和處理大容量科學數據設計的文件格式及相應庫文件

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

c. pip: 套件管理工具

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

d. 安裝相關套件

$ 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

2. Installing tensorflow-gpu

安裝完上述套件後，便可以開始安裝 tensorflow-gpu，如下:

$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-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: TensorFlow的每月NVIDIA容器版本，例如: 19.01。

3. 確認安裝成功

1. 開啟 Terminal ，並run:

$ python3

2. Import TensorFlow:

>>> import tensorflow

4 . 補充

一般來說在安裝python套件時都會使用下面命令:

$ sudo pip3 install 套件 

但是在Embedded linux下，常常會發現上面命令無法安裝相關套件

這時候可以使用 apt 來去安裝相關套件，如下:

$ sudo apt-get install python3-套件

以上簡單帶大家安裝 tensorflow-gpu

敬請期待下一篇 Jetson Nano的使用心得！

以上來源參考自:

https://docs.nvidia.com/deeplearning/dgx/install-tf-xavier/index.html