jesation nano部署yolov5-6.0

如遇问题，可私聊博主。

一.格式化SD卡，烧录镜像

二.提前在电脑上下载对应版本的torch和torchvision压缩包

官方网站为：PyTorch

进不去的时候科学上网，如有需要私聊博主。

三.在电脑上下载MobaXterm，jesation nano这块板子自身下载东西过于麻烦，利用软件远程操控和上传文件。（MobaXterm的使用方法在此不详细说明）

四.配置CUDA

sudo gedit ~/.bashrc

在最底下输入

export CUDA_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 ~/.bashrc
nvcc -V

正确配置后如下 

 

 查看Swap

free -h

 此时只有3.9G

输入 

sudo gedit /etc/systemd/nvzramconfig.sh

修改文件中1024改成4096 

保存后需要重启nano

sudo reboot

如果你t通过电脑重启nano失败，是权限的原因，自己手动重启nano 

再次查询Swap

free -h

五.更新pip 

 时间较久，在网络环境好的地方弄，不建议在板子上直接更新，通过Moba会快很多

sudo apt-get update
sudo apt-get upgrade
sudo apt-get dist-upgrade
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install --upgrade pip

六.安装torch

进入存放torch的文件夹，打开终端

pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install libjpeg-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
sudo apt install python3-numpy
pip install Cython
export OPENBLAS_CORETYPE=ARMV8

验证安装

python3
import torch 
print(torch.__version__)

七.安装torchvision

进入torchvision的文件夹

sudo apt-get install libjpeg8 libjpeg62-dev libfreetype6 libfreetype6-dev
export BUILD_VERSION=0.9.0
python3 setup.py install –user

验证安装

python3
import torchvision
print(torchvision.__version__)

八.安装yolov5-6.0

 上传yolov5-6.0

cd yolov5-6.0
pip install -r requirements.txt

到此，安装结束。