anaconda下Yolov5环境配置及常见问题处理
必备软件
默认已经安装好anaconda,没有安装的同学可以从以下网址安装:
清华镜像:https://mirrors.tuna.tsinghua.edu.cn/
1.虚拟环境创建
(1)进入anaconda prompt
(2) 创建虚拟环境
conda create -n yolov5500 python=3.8
python的版本根据你下载的yolo版本进行选择;
(3) 激活环境及配置清华源
conda activate yolov5500
(4) 配置清华源
配置清华源的目的在于加速后续相关安装包的下载;
1.打开以下网址:
https://mirrors.tuna.tsinghua.edu.cn/help/pypi/
2. 下滑找到以下代码:
复制第二行代码:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
3.回到终端(虚拟环境)界面,粘贴代码并执行
完成配置
2.安装Pytorch准备工作
1.确定电脑配置,打开英伟达控制面板
点击左下角系统信息:
确定cuda 12.2版本,之后安装pytorch cuda版本要小于12.2,并且要求适用于自己的显卡版本:
16XX的显卡安装cu102;
30XX、40XX显卡,安装cu111以上版本;
本次安装显卡1050TI;
3.安装Pytorch
1. 打开Pytorch官网
找到1.8.2版本
下滑找到,如下代码,并复制:
pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
2. 回到虚拟环境,粘贴代码并且执行
执行完成后输入以下代码:
python
import torch
torch.ones(3).cuda()
运行代码,得到以下结果,证明pytorch安装完成:
3.安装Yolov5
打开网站下载yolov5:
https://github.com/ultralytics/yolov5
点击Releases,下载源码,解压到你想要解压的目录:
打开解压文件中的requirements.txt:
修改以下四处,如下图
点击解压文件夹,输入cmd:
输入cmd,按Enter:
输入:
conda activate yolov5500
注意接下来的操作一定要关闭 弟子:
输入代码:
pip install -r requirements.txt
完成安装界面:
如果出现类似错误,很可能你没有关闭弟子或者代理处于打开状态,需要手动关闭
最后执行如下代码测试安装是否成功?
python detect.py
成功界面:
安装目录下找到runs\detect\exp2,显示如下图片,到此安装完成可以继续学习了~
##参考资料
大佬 “你可是处女座啊”: https://www.bilibili.com/video/BV1bg4y1R7cs/?spm_id_from=333.788&vd_source=e38ab283979a6c9e04d5b5069ed35b0d