windows Server2019配置yolov5环境

Anaconda

在清华镜像中下载Anaconda,我们下载的版本是
Anaconda3-2021.11-Windows-x86_64.exe 在这里插入图片描述之前将Anaconda安装到了D盘,要卸载干净后重新安装到了C盘下。安装时有一个添加进环境变量的配置要记得勾选。

cuda11.6

cuda官网默认的是11.7,为配合pytorch需要安装cuda11.6版本。
在这里插入图片描述cuda安装时点“自定义”即可。我们的默认是cuda17,没卸载,需要在用cuda 的时候标记cu116。
cuda版本查看:

nvidia-smi

在这里插入图片描述

pytorch安装步骤

根据cuda版本安装pytorch,必须Python版本+PyTorch版本+torchversion版本+torchaudio版本+cuda版本完全一一对应才能正常使用。
python版本是3.9,cuda是11.6,pytorch官网选择
在这里插入图片描述直接运行conmand报错,因为网络有些拉跨,就采用将后面的网址直接用浏览器打开,将原xommand中要安装的torch、 torchvision 和torchaudio依次下载

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

https://download.pytorch.org/whl/cu116

会看到一溜.whl文件, 挑一个版本下载。根据我们的配置,搜索“cu116-cp39-cp39‘,下载对应的版本。cu116就是cuda11.6,cp39就是python3.9.x缩写。
torchvision:
在这里插入图片描述torchandio:
在这里插入图片描述
torch:
在这里插入图片描述
下载好后,在whl文件的文件夹中打开命令行,运行

pip install 对应的whl文件的全称

测试:

python
>import torch
>torch.cuda.is_available()

返回True那么就是好了,可以跑显卡了。如果不行那么就是没安装对,只能用cpu跑。

yolov5代码

1.数据集存一个images、一个labels文件夹,labels转为txt格式
2.改一个coco.yaml,主要改:
a.train、val、test路径
在这里插入图片描述
b.改nc、names;nc表示数据集类别,那么时表示数据集标注的类名称
在这里插入图片描述
3.可以修改models模型文件,我们没改
4.训练train.py
weights、cfg、data按照文件路径修改。
从cfg默认初始models/yolovs.yaml,但我们训练的时候设为‘’
epochs地带次数自己决定,我们默认的是300
在这里插入图片描述5.训练中止时,exp/weights下会有best和last两个pt。继续开始训练只需要将last.pt路径写到weights下面/
在这里插入图片描述
6.测试时,只需要更改下面参数的路径即可
在这里插入图片描述

nice!

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

beryl2017

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值