1.安装yolov8的jetson版本docker
t=ultralytics/ultralytics:latest-jetson
sudo docker pull $t
2.docker run
sudo docker run -it --network host --runtime nvidia -v /home/agx/docker:/home/share --name train_yolov8 --shm-size=32g ultralytics/ultralytics:latest-jetson
–shm-size可以视情况更改
3.训练
yolo segment train data=data.yaml model=yolov8n-seg.pt epochs=100 batch=64