已经安装了python3 pip3
ubuntu18.04
jetpack4.5
cuda10.2
。。。。
1、安装PyTorch1.6.0
链接: https://pan.baidu.com/s/1c-N5JrWT0UbfratLNmzSiQ
提取码: evnm
百度网盘下载 torch-1.6.0-cp36-cp36m-linux_aarch64.whl文件,之后在.whl文件路径下运行:
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip install Cython
pip install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl
2、安装TorchVision0.7.0
下载TorchVision0.7.0版本文件夹:
sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.7.0 https://github.com/pytorch/vision vision-0.7.0
在vision-0.7.0路径下运行:(注意下载后源码要全部赋予读写权限)
sudo python setup.py install
3.安装torch2rt
apt-get install libopenblas-base libopenmpi-dev
pip3 install packaging -i https://pypi.tuna.tsinghua.edu.cn/simple
git clone https://github.com/NVIDIA-AI-IOT/torch2trt cd torch2trt (注意下载后源码要全部赋予读写权限) sudo python3 setup.py install