TX2 NX 部署GPU版本的yolov5

将 YOLOv5 的 GPU 版本部署到 Jetson TX2 NX 上需要进行以下步骤：

1. 首先，确保您的 Jetson TX2 NX 上已安装 CUDA 和 cuDNN。您可以从 NVIDIA 的官方网站下载并安装这些软件。

2. 接下来，您需要安装 Python3 和 pip3。您可以使用以下命令在 Jetson TX2 NX 上安装：

   sudo apt-get update
   sudo apt-get install python3-dev python3-pip

3. 安装 PyTorch 和 torchvision。您可以使用以下命令在 Jetson TX2 NX 上安装：

   sudo apt-get install libopenblas-base libopenmpi-dev 
   pip3 install Cython matplotlib
   pip3 install torch torchvision

4. 克隆 YOLOv5 仓库并切换到 v5.0 分支：

   git clone https://github.com/ultralytics/yolov5.git
   cd yolov5
   git checkout v5.0

5. 安装依赖项：

   pip3 install -r requirements.txt

6. 下载预训练模型：

   python3 models/download.py --weights yolov5s.pt

7. 运行 YOLOv5：

   python3 detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source 0

注意：在 Jetson TX2 NX 上运行 YOLOv5 可能需要一些优化。您可以尝试调整 batch size、input size 等超参数来提高性能。