站点图标 AI技术聚合

jetson nano 配置(远程、pytorch、yolov5等)

写在前面

虽然你们看我轻描淡写写了一千多个字,其实经历了两天两夜,而且我本人属于是安装Ubuntu四五遍的人,也玩过很长一段时间树莓派,在这个东西面前,还是研究了好一会,我觉得主要是硬件架构、nvidia魔改系统的问题(
总之,这件事最重要的是耐心。希望每个人都相信自己,敢于尝试,不惧怕从零开始,共创美好未来。

烧录固件

这个如果是国产的话可能要做,不同生产厂家有不同的烧录方式,问商家要就行。我是国产的使用一块16G emmc引导。fc rec这个引脚接地(和gnd短接),然后用商家烧写代码,让启动从tf卡开始。商家给了一个虚拟机,我真是醉了,为什么不直接给一个文件夹,一定要给虚拟机。

烧录系统

用etcher烧录,最好用最新版的

扩张

扩展应该是第一步,因为它可能会失败,然后一些以前做过的事情就会丢失。

安装gparted,找到sd卡(一般为/dev/sda)
右击已分配的分区,resize,拖到很靠右又不是全部在右边的地方(因为可能会报错),然后就可以开始了。如果失败了就只能重装啦。

连热点、装Nomachine

这个感觉还行Nomachine的包别找错了,应该是arm64

修改python版本的方法

设为3.6
参考https://www.cnblogs.com/ruiyang-/p/10162581.html

update-alternatives --install /usr/bin/python python /usr/bin/python2.7 1
update-alternatives --install /usr/bin/python python /usr/bin/python3.6 2
update-alternatives --config python

添加虚拟内存

如果在接下来的编译、安装等过程中不使用虚拟内存,可能会很慢,甚至会报错。


sudo fallocate -l 4G /var/swapfile
sudo chmod 600 /var/swapfile
sudo mkswap /var/swapfile
sudo swapon /var/swapfile
sudo bash -c 'echo "/var/swapfile swap swap defaults 0 0" >> /etc/fstab'

添加cuda相关路径

在~/.bashrc修改,差不多如下
我这样写

export CUBA_ROOT=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH

export CUBA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH

改完后可以source一下

安装pytorch和torchvision

一定要按照这个来,版本一定要对应,不要去试图安装通用包,跑不起来的,我安装了torch 1.10.0和torchvision 0.11.0a0(branch是0.11),是可以跑的(指yolo),其他的torchvision 比如0.11.3,就不可以了,会出现在这里插入代码片

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

最好命令前加个sudo

好像编译torchvision的时候会缺少Atomic.cuh(我只是少了这个),报错形如fatal error: ATen/cuda/Atomic.cuh: No such file or directory
去这里找相应的文件,可以在不同的版本里寻找,一般master里可以找到
https://github.com/pytorch/pytorch/tree/master/aten/src/ATen/cuda
放入/usr/local/lib/python3.6/dist-packages/torch/include/Aten/cuda
这个地方

配yolo相关的包

YOLO就是用原版的yolov5s

其他的包基本上都可以,好像matplotlib的安装有点麻烦,会遇到exit status -4: python setup.py egg_info这样的报错。安装前最好开虚拟内存,不然可能会卡在原地。
https://stackoverflow.com/questions/67588980/installing-matplotlib-error-command-errored-out-with-exit-status-4

pip install --upgrade pip setuptools wheel
pip install numpy==1.19.4
pip install matplotlib

在Stack Overflow看到了类似的情况,安装了下这个版本的numpy,后来好像就这样好了

此外

考虑用tensorrt加速,马一下
https://blog.csdn.net/carrymingteng/article/details/120978053

文章出处登录后可见!

已经登录?立即刷新
退出移动版