在阅读本章内容之前,需要把部署环境以及训练环境都安装好。
1.TensorRTX下载
这里使用Wang-xinyu大佬维护的TensorRTX库来对YOLOv8进行推理加速的演示,顺便也验证一下前面环境配置的成果。
github地址:GitHub - wang-xinyu/tensorrtx,下载后放到wsl的路径下,我这里放在/home路径下,并解压到/home路径下:
unzip tensorrtx-master.zip
2.Vs-Code远程连接TensorRTX
首先必须先安装一个WSL插件,如下所示:
然后连接WSL-Ubuntu2004,如下所示