@[TOC](Jetson Nano & TX2配置教程)
更新于2021年4月10日。本文介绍了Jetson Nano 和 Jetson TX2两种设备的配置教程,前一部分以Jetson Nano为例,后半部分以Jetson TX2为例。这两种设备的配置教程,大体上是相近的。
Jetson Nano 激活
首先,拿到一张64GB/128GB(12GB的卡不够用)的micro-SD card,并进行格式化。SD卡的格式化工具,直接从官网下载即可,使用时选择“快速格式化”。
然后,从Nvidia官网下载你需要的jetson nano镜像。
注意确定安装的jetpack版本,这关系到其自带的cuda版本。简单来说,jetpack4.4以上,只能安装pytorch1.6以上PyTorch for Jetson - version 1.8.0 now available。而且在pytorch1.6环境下保存的模型,没办法在更低版本的pytorch中打开。
旧版本可以从官网JetPack Archive寻找。
最后,根据Nvidia官网教程完成镜像的烧录,大概需要半个小时的时间。
Jetson TX2 烧录
这里给出两个链接,一个是官网链接使用SDK Manager安装Jetson软件,另一个是知乎上的链接Jetson TX2 入门教程(镜像烧写)。
更新系统国内源
在使用系统之前,我们先将系统更换为国内安装源。
首先,备份source.list。
1 | sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak |
然后,修改source.list。推荐用gedit,十分方便。如果没有安装的话,需要先安装一下。
1 | sudo apt-get install gedit |
接着,替换list中的内容。
1 | deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main multiverse restricted universe |
最后,更新软件。update和upgrade的区别可以参考linux命令系列 sudo apt-get update和upgrade的区别。
这一步可能需要花一段时间。
1 | sudo apt-get update |
安装系统工具jtop
Jetson Nano中有个工具jtop, 可以查看CPU和GPU资源,还可以显示系统的jetpack版本,十分好用。
首先,安装pip3.
1 | sudo apt-get install python3-pip |
同样的,我们也可以将pip修改为国内源。
第一种方式,长期修改。我以阿里源为例,其他国内源只需要修改链接即可。
1 | pip3 config set global.index-url https://mirrors.aliyun.com/pypi/simple/ |
然后,使用pip安装jetson-stats。
第二种方式,一次性修改。我以清华源为例,其他国内源只需要修改链接即可。
1 | sudo pip3 install jetson-stats -i https://pypi.tuna.tsinghua.edu.cn/simple |
安装好之后,我们来看一下效果如何。
1 | sudo jtop |
从源代码编译opencv
刷机后的Nano已经预装了opencv,但是预装版本不支持CUDA,所以我们还要在Jetson Nano上手动安装OpenCV。
方法一
安装已编译的opencv,简单方便,但是这个版本过低,且不能指定其他版本。
1 | sudo apt-get install opencv-python |
方法二
从源代码安装,但是使用别人写好的脚本,如buildOpenCVXavier,nano_build_opencv。但是报错之后,可能不知道自己为什么错了,因为脚本不是你自己写的。
方法三
自己从源代码编译安装,有很多坑要踩啊。
不能先安装archconda等虚拟环境管理包,否则会有问题!参考在Jetson Nano中为python3安装OpenCV
- 增加交换空间
opencv的编译时生成的中间文件很大,Jetson Nano原始空间放不下。这里介绍一种永久生效的方法。要是只想临时生效,可以参考Enlarge memory swap。
查看当前系统的交换空间。
1 | sudo swapon --show |
查看内存大小。
1 | free -h |
创建用于swap的文件。
1 | sudo fallocate -l 2G /swapfile |
设置交换空间。
1 | sudo mkswap /swapfile |
激活交换空间。
1 | sudo swapon /swapfile |
验证增加空间是否有效。
1 | sudo swapon --show |
- 安装依赖项
这个依赖项的要求五花八门,取决于你对功能的要求,这里我只给出我使用的依赖项,参考compile deeplearning libraries for jetson nano。如果遇到包冲突问题,参考aptitude与apt-get解决。
1 | dependencies=(build-essential |
- 下载源代码
安装所有第三方依赖项后,我们就需要下载OpenCV啦。需要两个压缩包:基本库和其他拓展库。在opencv官网下载你需要的版本。这里以4.4.0版本为例。
在github网址下载opencv-4.4.0.zip,复制下面的网址,在浏览器直接下载。
1 | https://github.com/opencv/opencv/archive/4.4.0.zip |
在github网址下载opencv_contrib-4.4.0.zip,复制下面的网址,在浏览器直接下载。
1 | https://github.com/opencv/opencv_contrib/archive/4.4.0.zip |
对下载后的opencv-4.4.0.zip和opencv_contrib-4.4.0.zip进行解压缩。
在opencv-4.4.0文件夹下打开命令行,新建一个build文件夹,并切换到build文件夹下。
1 | mkdir build |
- 编写cmake配置文件my_cmake.sh
这里是对opencv和cuda做的一些配置,和依赖库一样也是五花八门,看你的需求吧。这里我给出自己的配置,仅供参考。
新建脚本文件。
1 | gedit my_cmake.sh |
粘贴cmake命令。
1 | cmake -D CMAKE_BUILD_TYPE=RELEASE \ |
具体每个选项的介绍,可以参考Jetson Nano 从头配置OpenCV+CUDA+QT完整流程。
执行make命令。
1 | sh ./my_cmake.sh |
在我们进入实际编译步骤之前,请确保检查CMake的输出!参考树莓派安装OpenCV-4.1.0及Contrib
- 开始构建-j4表示4个CPU核心同时运行。
1
$ make -j4
查看CPU核心数,参考Ubuntu 18.04配置OpenCV 4.2.0。
1 | nproc |
经过漫长的等待,约两小时。
可能遇到错误,这是网络问题导致部分依赖包没有下载。
错误一
参考Jetson Nano 从头配置OpenCV+CUDA+QT完整流程中相关部分的解决方案。
1 | fatal error: boostdesc_bgm.i: No such file or directory |
错误二
1 | fatal error: opencv2/xfeatures2d/cuda.hpp: No such file or directory |
参考opencv安装opencv_contrib出现无法打开包括文件: “opencv2/xfeatures2d/cuda.hpp”中的解决方案。记得加在最开始,不要加在最后面!
错误三
1 | CMakeFiles/example_gpu_surf_keypoint_matcher.dir/surf_keypoint_matcher.cpp.o: In function `main': |
参考源码编译安装OpenCV4.5+ CUDA11 带python 遇到的错误总结中的解决方案。
- 安装
1 | make install |
- 验证
1 | $ python3 -c "import cv2; print(cv2.__version__)" |
安装archconda
anaconda是深度学习的好工具。在arm64框架平台上,对应有一个archconda,是大神编译好的工具,我们可以直接使用。
从大神github网址上下载sh文件,然后就像在普通linux上安装即可。
如果需要在虚拟环境中使用opencv,有两种方法。
方法一
链接,参考Installing OpenCV 4 on NVIDIA Jetson Nano。
1 | # Go to the folder where OpenCV's native library is built |
方法二
拷贝,参考How to check for successful install including in virtualenv. #32。
1 | # Go to the folder where OpenCV's native library is built |
在虚拟环境中导入cv2,可能会遇到下面一个错误。
1 | from .cv2 import * |
这主要是在编译opencv的时候使用的numpy,和虚拟环境中安装的numpy版本不一致,根据安装时的numpy版本修改一下即可,参考cv2: numpy.core.multiarray failed to import。
查看编译opencv的时候使用的numpy版本,可以进入系统自带的python环境,输出numpy版本号。
1 | >>> import numpy |
还有可能遇到另一个问题。
1 | >>> import numpy |
这是numpy版本的问题,将numpy=1.19.5降级到numpy=1.19.4或更低版本即可,参考Illegal instruction (core dumped)
安装mmdetection
这一部分使用的是Jetson TX2,参考在Jetson tx2安装 mmdetection环境。
这里我遇到了一个小问题,TX2接上网线但是连不上网,解决方法参考Jetson TX2 有线网络网线不识别,灯不亮问题。
其实,安装完opencv后,要运行mmdetection的代码,还有4个重要的包需要安装:torch,torchvision,mmcv-full,mmdet。
- 安装torch
去NVIDIA官网下载配套的版本PyTorch for Jetson - version 1.7.0 now available,这一步需要翻墙。
注意查看自己的Jetpack版本号,4.4要用torch1.6,否则会报错cudnn版本不合适。
查看Jetson设备的信息可以用之前提到的jtop,也可以使用查询Jetson设备与开发环境版本的基础信息。
下载好pytorch之后,激活虚拟环境,使用pip命令安装。
1 | pip install torch-1.6.0-cp36-cp36m-linux_aarch64.whl |
有可能遇到问题
1 | OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory |
参考Cannot install pytorch,安装两个依赖。
1 | sudo apt-get install libopenblas-base libopenmpi-dev |
- 安装torchvision
需要根据torch版本来选择对应版本的torchvision进行安装,参考Jetson Nano 安装torch和torchvision。
1 | git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision |
- 安装mmcv-full和mmdet
这里的教程跟mmdetection官方教程来走就好了。
安装mmcv-full
1 | git clone https://github.com/open-mmlab/mmcv.git |
安装mmdetection
1 | git clone https://github.com/open-mmlab/mmdetection.git |
注意:
- mmcv-full和mmdet的版本号要对应,官方教程已经给出来了对应关系。
- 安装mmcv-full,而不是mmcv,否则后面可能会有奇怪的错误。
1 | MMCV_WITH_OPS=1 pip install -e . -i https://pypi.tuna.tsinghua.edu.cn/simple |
- 在使用pip命令时,后面跟上国内源,这样下载其他依赖包会更快一些。
- 上面两个包安装完,也要花挺长时间的。
- 有可能遇到安装matplotlib,scipy,pillow等包的时候,与已经安装的numpy版本冲突。这种情况下不能修改numpy的版本,要修改其他包的版本。
- 可以使用下面这条命令,查看哪个版本的包时候你的numpy,包和版本号可以更改,参考conda查看某个安装包的依赖项。
1 | conda search scipy=1.0.0 -info |
- 测试
按照官网的代码,做了一些修改。
1 | from mmdet.apis import inference_detector, init_detector, show_result_pyplot |
参考
【Jetson-Nano】jetson_nano安装环境配置及tensorflow和pytorch安装教程
Jetson Nano配置与使用(1)开机
在Jetson Nano (TX1/TX2)上使用Anaconda与PyTorch 1.1.0
Install OpenCV 4.5.0 on Jetson Nano
compile deeplearning libraries for jetson nano
linux命令系列 sudo apt-get update和upgrade的区别buildOpenCVXavier
nano_build_opencv
在Jetson Nano中为python3安装OpenCV
Installing OpenCV 4 on NVIDIA Jetson Nano
Jetson Nano 从头配置OpenCV+CUDA+QT完整流程
树莓派安装OpenCV-4.1.0及Contrib
Ubuntu 18.04配置OpenCV 4.2.0
opencv安装opencv_contrib出现无法打开包括文件: “opencv2/xfeatures2d/cuda.hpp”
源码编译安装OpenCV4.5+ CUDA11 带python 遇到的错误总结
cv2: numpy.core.multiarray failed to import
在Jetson tx2安装 mmdetection环境