在Jetson Nano上安裝Pytorch與YOLOv5最新版6.x推理演示步驟

前提

你先得有個Jetson Nano的開發(fā)板，前提是準(zhǔn)備好SD卡！然后燒錄一個 jetpack4.6 版本以上的鏡像系統(tǒng)。下載鏡像到這里，推薦4.6版本直接下載：

https://developer.nvidia.com/embedded/l4t/r32_release_v7.1/jp_4.6.1_b110_sd_card/jeston_nano/jetson-nano-jp461-sd-card-image.zip

然后下載燒錄的工具：

https://www.balena.io/etcher/

完成燒錄之后，插到Jetson Nano 的SD卡槽中就可以通電啟動了，第一次啟動會看到！

01

安裝pytorch與torchvision

首先跟大家說一聲，一定要參考pytorch官方的文檔，它前面部分是非常有用的！文章地址：

https://pytorch.org/blog/running-pytorch-models-on-jetson-nano/

首先檢查版本，輸入命令行：

sudo apt-cache show nvidia-jetpack

然后安裝pip3命令行支持，因為后面安裝其他包需要，必須安裝！

sudo apt install python3-pip

安裝工具檢查CUDA版本

sudo pip3 install jetson-stats
sudo jtop

檢查一下tensorRT的版本：

pip3 list
or
python3
import tensorrt as trt
trt.__version__

檢測一下CUDA的各種信息

ls -lt /usr/local

下面就是安裝pytorch，首先去英偉達(dá)官方網(wǎng)站下載支持版本，這里筆者下載的是pytorch1.9版本。然后跟它對于的torchvision版本是0.10版本，這個也得自己下載。這兩個下載地址分別是：

https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl

https://github.com/pytorch/vision/archive/refs/tags/v0.10.0.zip

然后首先安裝下面兩個包，必須先安裝好！

pip3 install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install Cython

安裝1.9 pytorch，時間會長點，先摸魚一會再回來差不多就好了！

pip3 install torch-1.9.0-cp36-cp36m-linux_aarch64.whl

然后下載torchvision

https://github.com/pytorch/vision/archive/refs/tags/v0.10.0.zip

解壓縮，進入目錄，

unzip vision-0.10.0.zip
cd vision-0.10.0

先別著急直接安裝它，把這些依賴的包統(tǒng)統(tǒng)安裝一遍，然后會報錯！

sudo apt-get isntall libjpeg-dev zlib1g-dev
sudo apt install liblapack-dev
sudo apt install gfortran
sudo apt install libxft-dev libjpeg-dev libpng-dev

然后運行torchvision的setup.py開始安裝：

sudo python3 setup.py install

安裝完成之后，pip3 list檢查結(jié)果如下：

通過代碼查詢版本，可見已經(jīng)導(dǎo)入成功了，可以使用了。

02

安裝YOLOv5 新版本6.x與測試

先什么都別說拉，把這兩個安裝一波！

pip3 install -–no-cache-dir pillow -i https://pypi.tuna.tsinghua.edu.cn/simple

pip3 install seaborn -i https://pypi.tuna.tsinghua.edu.cn/simple

然后下載YOLOv5 6.1版本

https://github.com/ultralytics/yolov5/archive/refs/tags/v6.1.zip

下載之后，解壓縮，進入目錄：

unzip yolov5-6.1.zip
cd yolov5-6.1

然后在命令行直接運行下面得腳本：

pythondetect.py--weightsyolov5s.pt--sourceimages/data/zidane.jpg--view-img

正常情況下是可以直接運行了，如果發(fā)生一些python module找不到得情況，比如：

The ModuleNotFoundError: No module named ‘yaml’

直接：

pip3 install pyyaml

安裝好之后繼續(xù)執(zhí)行上面得檢測腳本，直到成功為止。最終運行成功顯示如下：

視頻文件也可以得：

然后我把一個自定義訓(xùn)練好的模型，部署到nano上去了，發(fā)現(xiàn)也是可以直接推理，顯示如下：

審核編輯：劉清