1. 安装pytorch

sudo apt install nvidia-cuda-toolkit
nvcc --version 
# 官方适配地址：https://download.pytorch.org/whl/torch/

import torch
print(torch.__version__)
print(torch.cuda.is_available())

2. 安装环境

# cuDNN 安装：https://developer.nvidia.com/rdp/cudnn-archive
# Anaconda 安装：https://www.anaconda.com/download
sh Anaconda3-2024.10-1-Linux-x86_64.sh

3. 安装yolo

pip install ultralytics

mkdir myYolo_test && cd myYolo_test
yolo task=detect mode=predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'