Jetson设备配置yolo环境

sudo apt install python3-pip libopenblas-base libopenmpi-dev libomp-dev libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev

pip config --user set global.index-url https://mirrors.huaweicloud.com/repository/pypi/simple
pip install --upgrade pip

# https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/

wget https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

pip install torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

git clone --branch release/0.20 https://github.com/pytorch/vision torchvision

cd torchvision
export BUILD_VERSION=0.20
python3 setup.py install --user # 会报错，需要安装cusparselt：https://developer.nvidia.com/cusparselt-downloads

# install cusparselt
wget https://developer.download.nvidia.com/compute/cusparselt/0.9.1/local_installers/cusparselt-local-repo-ubuntu2204-0.9.1_0.9.1-1_arm64.deb
sudo dpkg -i cusparselt-local-repo-ubuntu2204-0.9.1_0.9.1-1_arm64.deb
sudo cp /var/cusparselt-local-repo-ubuntu2204-0.9.1/cusparselt-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cusparselt

# 再次尝试安装
python3 setup.py install --user

import torch
print('CUDA available: ' + str(torch.cuda.is_available()))
print('cuDNN version: ' + str(torch.backends.cudnn.version()))
a = torch.tensor([0., 0.], dtype=torch.float32, device='cuda')
print('Tensor a =', a)
b = torch.randn(2, device='cuda')
print('Tensor b =', b)
c = a + b
print('Tensor c =', c)

import torchvision
print(torchvision.__version__)

pip uninstall numpy
pip install numpy==1.26.4

pip install ultralytics

yolo export model=best.pt format=engine device=0 half=True imgsz=640 simplify=False workspace=4

sudo pip install jetson-stats
sudo systemctl enable jtop.service
sudo systemctl start jtop.service

sudo apt install -y \
  python3-gi \
  python3-gst-1.0 \
  gstreamer1.0-tools \
  gstreamer1.0-plugins-base \
  gstreamer1.0-plugins-good \
  gstreamer1.0-plugins-bad \
  gstreamer1.0-plugins-ugly \
  libgstreamer1.0-dev \
  libgstreamer-plugins-base1.0-dev \
  gstreamer1.0-plugins-base-apps

pip uninstall opencv-python opencv-contrib-python opencv-python-headless

sudo apt install -y python3-opencv