1.安装python以及依赖包
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git cmake python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev
备注:上面的步骤一步一步执行
2.配置python环境
sudo apt-get install python3-pip
sudo pip3 install -U pip testresources setuptools
3.安装Pytorch:
sudo apt-get install libopenblas-base libopenmpi-dev
sudo pip3 install mpi4py
sudo pip3 install Cython
备注:先下载好pytorch编译好的.whl文件,在Jetson Nano部署主要难点是pytorch安装,使用Nvidia官方预编译whl文件.选择自己需要的版本下载.
下载链接:PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums
不需要解压,下载完成拷贝到NX后,进行安装,注意文件存放位置和执行命令的位置相一致,输入命令:
sudo pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl
4.安装torchvision
进入其官方GitHub:GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
找到自己需要的版本下载下来,torch和torchvision的版本需要相互对应,比如博主最终安装的是Pytorch1.7.0,所以torchvision版本选择0.8.
下载完成后,解压,进入解压后的目录:
在此处打开终端,输入:
peter@peter-desktop:~/Desktop/jetson-nano-fire-detection-master/vision-0.8.0$ sudo python3 setup.py install
如果没有报错,那么恭喜你,安装成功啦!
5.[避坑总结]torchvision安装报错
博主查阅诸多资料,看了很多的帖子都没有找到合适的解决方法 ,最后终于找到了一篇帖子完美的解决了博主的问题,在此非常感谢!
通过阅读报错的地方发现了问题所在,这一条报错信息:
/home/peter/Desktop/jetson-nano-fire-detection-master/vision-0.8.0/torchvision/csrc/cpu/decoder/defs.h:12:10:
指向/home/peter/Desktop/jetson-nano-fire-detection-master/vision-0.8.0/torchvision/csrc/cpu/decoder/defs.h
文件
根据这个地址找到错误出现的地方:
出现这个错误其实就是缺少这些包嘛,只需要去下载这些包就可以完美的解决问题啦!
sudo apt install libavcodec-dev
sudo apt install libavformat-dev
sudo apt install libswscale-dev
再到vision文件夹下打开终端,然后在终端中输入
sudo python3 setup.py install
在执行这一行代码时发现pillow包一直没有安装成功,其实到这里torchvision已经安装成功了,输入命令pip3 list可以看到。只需要pip3 install pillow安装pillow包即可,如果下载速度慢,可以加入-i加清华源下载。
peter@peter-desktop:~/Desktop/jetson-nano-fire-detection-master/vision-0.8.0$ pip3 install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple
安装成功啦!