jetson nano 安裝 tensorflow
(1)使用 jtop 查看 Jetpack 版本號
刷新 apt 存儲庫索引
$ sudo apt-get update
使用 apt-get 安裝 pip3
$ sudo apt-get install python3-pip
使用 pip3 安裝 jtop,jtop 是一個用於監測和控制 NVIDIA Jetson 系列產品的軟件,雖然在終端運行,但是界面做的非常漂亮。你可以用它監測 CPU, GPU, 風扇等的運行情況。
jtop 的 github 鏈接: https://github.com/rbonghi/jetson_stats
$ sudo -H pip3 install -U jetson-stats
啟動 jtop,如果不成功的話可能需要重啟
$ jtop
按數字鍵 1~6 可以跳轉到不同的控制界面,按 q 退出
在界面 6 [INFO] 一欄可以看到 Jetpack 版本信息,我的機器是 4.5.1 版本的
或者使用 jetosn_release 指令也可以查看機器信息:
$ jetson_release -v
同樣可以看到,Jetpack 是 4.5.1 版本的。
(2)從官網下載並安裝 Tensorflow
NVIDIA 在 Jetson Nano 上安裝 Tensorflow 的官方教程網址為: Official TensorFlow for Jetson Nano! - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums
因為我已經確認了我的機器上的 Jetpack 為 4.5 版本的,所以我按照第一部分即(Python 3.6 + Jetpack 4.5)下面的教程操作,依次輸入以下指令(建議直接去官網上找,因為具體內容可能隨着時間會不斷更新):
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
sudo apt-get install python3-pip
sudo pip3 install -U pip testresources setuptools==49.6.0
sudo pip3 install -U numpy==1.16.1 future==0.18.2 mock==3.0.5 h5py==2.10.0 keras_preprocessing==1.1.1 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11
# TF-2.x
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v45 tensorflow
# TF-1.15
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v45 ‘tensorflow<2’
后兩步可能比較慢,耐心等待。我沒有更換 pip3 的源也都下載安裝成功了,如果因為網絡原因失敗了的話也可以換一下 pip3 的源再嘗試。
(3)檢查安裝是否成功
進入 python 交互環境:
$ python3
嘗試 import tensorflow:
>>> import tensorflow as tf
查看版本:
>>> tf.__version__
出現以下界面表示成功
(4)可能遇到的問題及解決方法
我遇到了 import tensorflow 指令不成功的錯誤,報錯 Illegal instruction(core dumped),import numpy 也是一樣的報錯,而且重啟、重裝系統問題依舊。
解決方法:打開 ~/.bashrc 文件,在末尾添加一行 export OPENBLAS_CORETYPE=ARMV8
此時再按照上一步檢驗 tensorflow 版本,就沒有問題了。
(5)參考文章
[1] Jetson Nano 入門教程3 - 必備軟件安裝Pytorch TensorFlow - 知乎 (zhihu.com)
[3] 解決英偉達Jetson平台使用Python時的出現“Illegal instruction(cpre dumped)”錯誤_簡明AI工作室-CSDN博客
