1、创建一个虚拟环境
首先去自己的anaconda的安装的envs(虚拟环境),在导航栏输入cmd,进入命令窗口
确保python>=3.7;CUDA>=10.1,PYtorch>=1.7
conda create -n torch1.12.1 python=3.8.8
2、激活刚建的虚拟环境
activate torch1.12.1
3、到官方网站下载yolo模型 ,下载好后解压,里面有个文件requirements.txt
mirrors / ultralytics / ultralytics · GitCode
安装一个整体包:
pip install -r .\requirements.txt
直接按照路径会有问题,找到自己 requirements.txt 文件路径,我这里是pip install -r D:\ultralytics-main\ultralytics-main\requirements.txt
4、然后安装ultralytics ,这是必须的。可以用镜像地址
pip install ultralytics -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
5、安装下载好包,接下来就是验证:
yolo predict model=yolov8n.pt source='ultralytics/assets/bus.jpg' show=True save=True