(3)torch2trt等安装

该博客详细记录了在Jetpack 4.5.1环境下安装torch2trt的过程,包括安装opencv-python、PyTorch 1.10、torchvision 0.11.1,以及解决依赖问题。还提供了手动下载torchvision源码并编译的步骤,并指导如何安装libopenblas和libopenmpi-dev。最后介绍了安装torch2trt的命令和检查安装成功的测试方法。
部署运行你感兴趣的模型镜像

为了把torch2trt装上,切换会jetpack4.5.1

主要参考了

https://blog.youkuaiyun.com/chencaw/article/details/117902535
https://blog.youkuaiyun.com/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的对应关系如下:

torchvision · PyPI

 

所以对应版本为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__)
————————————————
版权声明:本文为优快云博主「chencaw」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.youkuaiyun.com/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

您可能感兴趣的与本文相关的镜像

Python3.9

Python3.9

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值