torch2trt:
https://github.com/NVIDIA-AI-IOT/torch2trt
安装参考指令:
git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
sudo python setup.py install --plugins
如果没有PyTorch环境,请参考:
PyTorch for Jetson - version 1.6.0 now available
http://www.10qianwan.com/articledetail/635508.html
参考我的Jetpack版本(4.4),应该安装:torch-1.6.0-cp36-cp36m-linux_aarch64.whl
下载了离线文件,安装指令:
pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
之后边便是验证,使用。
版权声明:本文为weixin_41783910原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。