一、下载安装Miniconda
通过清华镜像源下载Miniconda
下载地址:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_22.11.1-1-Windows-x86_64.exe
下载完毕后点击安装
注意:这里最好勾选下添加到环境变量
以win11为例:按住win键,在搜索栏输出关键词,出现Anaconda,点击"打开"
conda create -n yolov5 python=3.8
打开后会显示如下界面:
在输入框中输入:
conda create -n yolov5 python=3.8
出现此语句:输入y
进入yolov5的包
conda activate yolov5
二、配置pypi国内镜像源
切换清华源
python -m pip install --upgrade pip
pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
三、安装Pytorch
进入官网:PyTorch
注意:自己电脑的CUDA版本
查看CUDA版本(以win11为例)
右键选择此图标,选择"NVIDIA控制面板"
点击左下角的"系统信息",在系统信息界面选择"组件",即可查看CUDA版本,我的CUDA最大支持版本是12.3
这里我们安装一个稳定版本——1.8.2
在历史版本中:Previous PyTorch Versions
找到
根据自己的CUDA版本,选择合适的版本(AMD选择CPU运行)
根据我的电脑配置,在运行命令框中,选择以下命令开始安装
pip install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu102
安装包比较大,耐心等待安装完毕
三、安装yolov5
下载地址:https://github.com/ultralytics/yolov5
选择本地下载,并将此文件,解压到一个没有中文目录的文件中
打开yolov5-master后,找到requirements.txt,里面可以看到yolov5需要安装的包
在命令指令框中输入
pip install -r requirements.txt
安装完毕后,运行测试代码
python detect.py
在安装过程中,可能会卡在这一步
可以直接到这个网址下载yolov5s.pt,放在yolov5-master这个文件下即可
之后,重新运行
python detect.py
运行成功后,打开\yolov5-master\runs\detect\exp
这里有两张测试图像,至此yolov5环境配置成功