前言
提示:安装的东西比较多,需要有耐心的往下看
安装要求:
(1)miniconda
(2)pytorch
(3)ultralytics
一、miniconda
miniconda相较于anaconda更加的轻量化下载地址
最好安装在C盘
然后点击右下角install就可以了
二、创建conda虚拟环境
打开下载好的conda
输入命令
conda create -n 起个名字 python=3.8
注意:这里我们需要指定一下python的版本
这里输入y,表示允许继续创建
这样表示创建成功,然后我们需要激活一下已经创建好的环境
conda activate 你自己起的名字
创建成功
配置镜像源
如果不配置的话会下载的很慢,输入命令
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
三、pytorch安装
推荐官网:下载网址
这里需要看一下自己的配置,电脑右键选择nvidia控制面板
然后点击左下角的系统信息
选择组件,查看一下自己的cuda,我这里的是12.1,所以我安装的版本低于12就可以了
另外如果是:
16–的显卡,需要安装cu102版本,否则训练会出问题
30–或40–比较新的显卡,要安装cu111以上的版本,否则无法运行
不同配置选择不同版本,我这里选择的是
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
显示安装成功
四、安装ultralytics(yolov8)
需要强调的是,这时已经是激活了自己的环境了,需要在自己的环境内操作,首先需要cd进入你的ultralytics-main目录下,然后执行命令
pip install -e .
最后输入命令
yolo predict model=yolov8n.pt source=ultralytics/assets/bus.jpg
(如果没报错可以忽略这步)这里如果显示yolov8n.pt下载失败,我们可以手动下载这个文件,然后放在ultralytics-main目录下:
下载链接
重新运行上面的命令,显示以下就是运行成功了
检测到一个bus,4个人等等…