参考资料: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'
近期评论