先写最重要的几个步骤
本机环境如下
一、安装tensorrt
1、到这里选择合适的下载NVIDIA TensorRT Download | NVIDIA Developer(我下载的是这个Log in | NVIDIA Developer和我出场自带的版本一样)
2、下载完成后按照下面图片所示的目录选择合适的whl安装
(我的python为3.8安装的第三个)
二、安装pytorch
1、找合适的pytorch
链接奉上:PyTorch for Jetson - Jetson & Embedded Systems / Announcements - NVIDIA Developer Forums
注意:要看jetpack版本列如我的jetpak为5.0.2那么我就可以安装以下这些
2、安装pytorch
下载完成后直接pip安装即可(我安装的是1.12.0)
三、安装torchvision
1、在这里找一下刚刚