参考资料:https://blog.csdn.net/weixin_43947712/article/details/115530913

一、下载yolov5

git clone git://github.com/ultralytics/yolov5.git

二、安装其它相关库

pip3 install matplotlib pillow pyyaml tensorboard tqdm scipy
sudo apt-get install python3-matplotlib

三、torch环境配置

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

四、torchvision环境配置

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.9.0  # where 0.x.0 is the torchvision version  
python3 setup.py install --user
cd ../
pip install 'pillow<7'