diff --git a/YOLOv5-3.X.md b/YOLOv5-3.X.md index 92ea64b..ed8230b 100644 --- a/YOLOv5-3.X.md +++ b/YOLOv5-3.X.md @@ -25,18 +25,19 @@ sudo apt-get install python3-matplotlib * PyTorch (for Jetson plataform) ``` -wget https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl -O torch-1.6.0-cp36-cp36m-linux_aarch64.whl +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 torch-1.6.0-cp36-cp36m-linux_aarch64.whl +pip3 install Cython +pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl ``` * TorchVision (for Jetson platform) ``` -git clone -b v0.7.0 https://github.com/pytorch/vision torchvision -sudo apt-get install libjpeg-dev zlib1g-dev python3-pip +sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev +git clone --branch 0.9.0 https://github.com/pytorch/vision torchvision cd torchvision -export BUILD_VERSION=0.7.0 -sudo python3 setup.py install +export BUILD_VERSION=0.9.0 +python3 setup.py install --user ``` ##