边缘计算机jetson AGX orin配置pytorch和YOLO11使用

一、安装环境

        1、检查系统环境

        检查系统环境，一般都已经安装好jetpack了，所以我们只需要查询jetpack版本跟系统环境信息执行 cat /etc/nv_tegra_release 和 sudo apt-cache show nvidia-jetpack 查看。

        2、下载安装pytorch

   进入nvidia网址（https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048）安装适配的 pytorch-gpu版本（不建议安装cpu，基本上不能应用ai）。例如本机使用的 jetpack 5.1.2，选择安装 PyTorch v2.1.0 版本即可。

既然找到网址了我一般是下载好whl文件然后本地下载，下载好了之后直接pip下载就行。

pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

安装后，验证

Python

import torch

可能出现的错误，和解决办法

ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory

sudo apt-get install libopenblas-base

3、下载torchvision

去torchvision的github网址查看版本要求，

因为我的torch是2.1，所以选择0.16的torchvision下载，

git clone --branch v0.16.1 https://github.com/pytorch/vision torchvision

export BUILD_VERSION=0.16.1

python setup.py install --user

编译后验证，

python

import torchvision

我遇到的错误是

image.py:13: UserWarning: Failed to load image Python extension: ''If you don’t plan on using image functionality from torchvision.io, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have libjpeg or libpng installed before building torchvision from source？

就是说我在编译torchvision前没有安装libjpeg跟libpng，所以我们就安装这两。执行sudo apt-get install libjpeg-dev libpng-dev zlib1g-dev之后，删除所有缓存和编译零时文件，再重新编译安装即可。

1. 测试安装是否成功

python

import torch

print(torch.__version__)

print('CUDA available: ' + str(torch.cuda.is_available()))

print('cuDNN version: ' + str(torch.backends.cudnn.version()))

a = torch.zeros(2, device='cuda')

print('Tensor a = ' + str(a))

b = torch.randn(2, device='cuda')

print('Tensor b = ' + str(b))

c = a + b

print('Tensor c = ' + str(c))



import torchvision

print(torchvision.__version__)

若均不报错，且能正常输出说明安装成功。

二、安装ultralytics

1、找到YOLO代码相关路径

因为我是把代码放u盘里，所以我们首先得知道我们的u盘位置，然后再进入到代码目录，

lsblk

cd /media/nvidia/PS2000

cd ultralytics-main

2、下载配置文件

YOLOv8或者之前的是有一个requirement.txt但是后面版本的YOLO就比如YOLOv11的配置文件是pyproject.toml，所以我们不能pip install requirement.txt,我们应该使用pip install pyproject.toml,或者懒得找的话就直接使用pip install .下载配置文件，pip install . 这条命令的意思是从当前目录（即 . 所代表的当前路径）安装 Python 项目。

具体来说，pip 会在当前目录下查找并解析 Python 项目的配置文件，如 setup.py 或 pyproject.toml，然后根据这些配置文件来安装该项目及其依赖项，所以，我们得确保你所在目录上有setup.py 或 pyproject.toml。

三、YOLO使用

1、转到对应python文件路径

cd ultralytics

2、运行YOLO预测

python predict-detect.py

带界面的话得额外下一个pyqt5，

sudo apt-get install python3-pyqt5

然后运行自己写好的界面代码就好，运行结果如图所示，