Yolov10的环境搭建与demo运行
下载yolov10源码
https://github.com/THU-MIG/yolov10
论文:https://arxiv.org/pdf/2405.14458
一、首先确认自己的运行环境
我的:
Ubuntu 22.04.2 LTS
查询cuda版本
nvcc -V
我的:Cuda compilation tools, release 11.7, V11.7.99
或者:nvdia-smi
二、创建虚拟环境
conda create -n yolov10 python=3.10
conda activate yolov10
在pytorch的官网上,找到适合自己系统的torch的版本。
Pytorch官网:https://pytorch.org/
我的:可以安装torch 2.0.1版本,使用官方命令安装如下
conda install pytorch