一 环境配置
1.1 Jetson平台的GPU版本torch与torchvison的安装
一般在使用YOLOv8时候,通常由两种模式:
1)pip install ultralytics
:不需要修改模型的话,直接用这个方便
2) pip installl requirement.txt -r
:对模型结构修改的话只能用这个
作者是为了对模型进行改进,并测试其性能,使用方式2
-
安装完毕之后,把torch与torchvision卸载掉,使用
uninstall
命令 -
下载支持GPU版本的touch与torchvision,PyTorch for Jetson - Jetson & Embedded Systems / Announcements - NVIDIA Developer Forums参考网址
备注:torch如果没有你对应平台的版本,可以自己自行进行编译,torchvision一般都是自己构建的.这里可以参考这篇文章jetson NX部署Yolov8_jetson部署yolov8-CSDN博客
安装完成之后
安装完成之后,可以使用predict函数测试一下一张图片&