Windows下Anaconda中配置Yolov5环境

Anaconda安装pytorch

1、换源

增加下载速度,换源到国内镜像源,在cmd控制台或者powershell下输入命令即可

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes

2、创建虚拟环境

打开Anaconda Prompt,此时应该是在(base)环境下,直接输入

conda create -n pytorch python=3.7

创建了一个名为pytorch的虚拟环境,环境中python版本为3.7
以后再打开Anaconda Prompt, 命令行输入

conda activate pytorch 

可直接进入虚拟环境。

3、安装pytorch

直接进入pytorch官网,附上连接:pytorch官网地址
官网页面如下,需要注意:30系列显卡需要装CUDA11以上的版本!!
官网页面
由于之前换了源,直接在命令行输入一下命令,不用-c pytorch ,这个是从官方地址下载,国外会比较慢。

conda install pytorch torchvision torchaudio cudatoolkit=11.3

安装时间会稍微长一些,不用操作等一会儿就安装好了。

4、下载Yolov5

附上官方地址,我下的是Yolov5-v5.0版本的

https://github.com/ultralytics/yolov5

Github主页,可以自选版本。
在这里插入图片描述
按照项目文件里的requirement.txt安装所需所需包。

pip install -r requirement.txt

pycocotools包会出现问题。
找下面这位兄弟。他已经共享了。
https://blog.csdn.net/oijdkd/article/details/120344665

欢迎大家点赞收藏,后面有空再介绍利用TensorRt在Jetson嵌入式系列板子上的部署。

Windows 10 上配置 YOLOv5 环境需要以下步骤: 1. 安装 Anaconda:从官网下载并安装 Anaconda,选择 Python 3.x 版本。 2. 创建虚拟环境:打开 Anaconda Prompt 终端,输入以下命令创建一个名为 yolov5 的虚拟环境: ``` conda create --name yolov5 python=3.8 ``` 3. 激活虚拟环境:在终端输入以下命令激活 yolov5 虚拟环境: ``` conda activate yolov5 ``` 4. 安装 PyTorch:在 yolov5 虚拟环境,输入以下命令安装 PyTorch: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia ``` 如果没有 GPU,可以使用以下命令安装 CPU 版本: ``` conda install pytorch torchvision torchaudio cpuonly -c pytorch ``` 5. 下载 YOLOv5 代码:在 yolov5 虚拟环境,输入以下命令下载 YOLOv5 代码: ``` git clone https://github.com/ultralytics/yolov5.git ``` 6. 安装依赖库:在 yolov5 文件夹,输入以下命令安装依赖库: ``` cd yolov5 pip install -r requirements.txt ``` 7. 测试 YOLOv5:在 yolov5 文件夹,输入以下命令测试 YOLOv5: ``` python detect.py --source 0 # 使用摄像头进行实时检测 ``` 如果想要使用预训练模型进行检测,可以使用以下命令: ``` python detect.py --weights yolov5s.pt --img 640 --conf 0.4 --source data/images/bus.jpg ``` 其,--weights 参数指定模型权重文件,--img 参数指定输入图像大小,--conf 参数指定置信度阈值,--source 参数指定输入图像路径或摄像头编号。 以上就是在 Windows 10 上配置 YOLOv5 环境的步骤。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值