1.Anaconda 创建虚拟环境
conda create -n Yolov10 python=3.8 -y
2. 安装ultralytics
pip install ultralytics -i https://pypi.mirrors.ustc.edu.cn/simple
3.安装Yolov10.git
pip install -q git+https://github.com/THU-MIG/yolov10.git
4.安装pytorch
如果是GPU,安装适合CUDA11.8的pytorch(确保已经按照CUDA和CUDNN对应版本)
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia
如果是想运行CPU,
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 cpuonly -c pytorch