TX2安装torch2trt

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版权协议,转载请附上原文出处链接和本声明。