为了把torch2trt装上,切换会jetpack4.5.1
主要参考了
https://blog.csdn.net/chencaw/article/details/117902535
https://blog.csdn.net/m0_51004308/article/details/114747285
(1)安装opencv-python
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
(2)装pytorch的地址
PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums
(3)安装torchvision
(1)
pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
(2)如果装不上torch和torchvision的对应关系如下:
所以对应版本为0.11.1版本
试试
pip install torchvision==0.11.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
(3)ru guo hai buxing
https://github.com/pytorch/vision
手动选择下载0.11.1 banben
解压后,进入文件夹
export BUILD_VERSION=0.11.1
sudo python3 setup.py install
一定要耐心等待,如果等不了,可以通过jtop看看情况
jtop
中间用到pillow>=4.1.1如果比较慢建议先安装
pip3 install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com pillow
#测试是否安装成功
python3
import torchvision
print(torchvision.__version__)
————————————————
版权声明:本文为CSDN博主「chencaw」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/chencaw/article/details/117902535
(4)anzhuan torch2rt
apt-get install libopenblas-base libopenmpi-dev
pip3 install packaging -i https://pypi.tuna.tsinghua.edu.cn/simple
ranhou
git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
sudo python setup.py install --plugins