thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated PyTorch and Torchvision for Jeston

Browse Source 
PyTorch 1.8.0
Torchvision 0.9.0
This commit is contained in:
Marcos Luciano
2021-05-09 16:54:35 -03:00
committed by GitHub
parent 6e8b9a97fa
commit 9e2a2d9355
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
YOLOv5-3.X.md
View File

@@ -25,18 +25,19 @@ sudo apt-get install python3-matplotlib
* PyTorch (for Jetson plataform) * 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 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) * TorchVision (for Jetson platform)
``` ```
git clone -b v0.7.0 https://github.com/pytorch/vision torchvision sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install libjpeg-dev zlib1g-dev python3-pip git clone --branch 0.9.0 https://github.com/pytorch/vision torchvision
cd torchvision cd torchvision
export BUILD_VERSION=0.7.0 export BUILD_VERSION=0.9.0
sudo python3 setup.py install python3 setup.py install --user
``` ```
## ##