jetson nano 安裝 tensorflow


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 退出

jtop

在界面 6 [INFO] 一欄可以看到 Jetpack 版本信息,我的機器是 4.5.1 版本的

或者使用 jetosn_release 指令也可以查看機器信息:

$ jetson_release -v

jetson_release

同樣可以看到,Jetpack 是 4.5.1 版本的。

(2)從官網下載並安裝 Tensorflow

NVIDIA 在 Jetson Nano 上安裝 Tensorflow 的官方教程網址為: Official TensorFlow for Jetson Nano! - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums

nvidia

因為我已經確認了我的機器上的 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__

出現以下界面表示成功

tensor

(4)可能遇到的問題及解決方法

我遇到了 import tensorflow 指令不成功的錯誤,報錯 Illegal instruction(core dumped)import numpy 也是一樣的報錯,而且重啟、重裝系統問題依舊。

解決方法:打開 ~/.bashrc 文件,在末尾添加一行 export OPENBLAS_CORETYPE=ARMV8

此時再按照上一步檢驗 tensorflow 版本,就沒有問題了。

(5)參考文章

[1] Jetson Nano 入門教程3 - 必備軟件安裝Pytorch TensorFlow - 知乎 (zhihu.com)

[2] Official TensorFlow for Jetson Nano! - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums

[3] 解決英偉達Jetson平台使用Python時的出現“Illegal instruction(cpre dumped)”錯誤_簡明AI工作室-CSDN博客

[4] Illegal instruction (core dumped) on import for numpy 1.19.5 on ARM64 · Issue #18131 · numpy/numpy (github.com)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM