网站设计风格确认书,php网站链接支付宝,国办网站建设规范,做酒店销售上哪个网站好目录 下载 CUDA 下载 cuDNN 下载 anaconda 安装 PyTorch pycharm 搭配 yolo 环境并运行 阅读本文须知#xff0c;需要电脑中有 Nvidia 显卡
下载 CUDA
打开 cmd #xff0c;输入 nvidia-smi #xff0c;查看电脑支持 CUDA 版本#xff1a; 我这里是12.0#xff0c;进入… 目录 下载 CUDA 下载 cuDNN 下载 anaconda 安装 PyTorch pycharm 搭配 yolo 环境并运行 阅读本文须知需要电脑中有 Nvidia 显卡
下载 CUDA
打开 cmd 输入 nvidia-smi 查看电脑支持 CUDA 版本 我这里是12.0进入 CUDA 官网https://developer.nvidia.com/cuda-toolkit-archivehttps://developer.nvidia.com/cuda-toolkit-archive
找到对应版本 按照自己电脑配置选择点击下载 双击启动选择下载位置 注意自定义安装才能选择安装位置 一路点击下一步继续等等完成下载。
安装出现问题的可用看看这个
【安装完CUDA后文件夹消失解答】_cuda安装后文件夹不见了-CSDN博客
是否下载成功输入 nvcc -V 下载 cuDNN
下载完成后为了实现加速还需下载对应的 cuDNN
cuDNN Archive | NVIDIA Developer
这里以我为例 下载解压完是三个文件夹 将这三个文件夹中的内容分别复制到之前下载的 CUDA 中对应的问价夹中。
环境配置
点击【系统】--【高级系统设置】--【环境变量】--系统变量中的【Path】--【新建】 新建刚才添加的三个文件 验证 下载 anaconda
大家可用自行下载注意下载时要勾选上添加环境变量
打开 anaconda 命令行 输入 conda create -n YOLO-GPU python3.8 验证环境
先后执行下面脚本 conda activate YOLO-GPUpython安装 PyTorch
选择与 cuda 对应的 pytorch 版本。安装的 cuda 版本需大于等于 pytorch 支持的版本。
Start Locally | PyTorch 官网上已经不推荐使用 Conda 下载了可用选择使用 Pip 下载Conda 下载命令这里也给出在 anaconda prompt 中依次输入
conda activate YOLO-GPUconda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidiapip 的下载命令官方给的速度太慢可以加上清华源下载
pip install torch torchvision torchaudio --index-url https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu121 百度网盘 请输入提取码百度网盘为您提供文件的网络备份、同步和分享服务。空间大、速度快、安全稳固支持教育网加速支持手机端。注册使用百度网盘即可享受免费存储空间https://pan.baidu.com/s/1dxxndKEj607g1wlkl6mgSg?pwdknvg#list/path%2Fsharelink3901291660-1103934630550953%2FYOLOv8%2FYOLO-GPU%E7%89%88%E4%B8%93%E7%94%A8%E5%8C%85parentPath%2Fsharelink3901291660-1103934630550953 实在不行可以直接用上面百度网盘或者我资源里的将那三个文件放在某个地方放哪儿都行使用 Anaconda Navigator打开 YOLO-GPU 终端 定位到存放前面三个文件的地方然后依次输入以下三个命令将这三个包下载到虚拟环境中
pip install torch-1.13.1cu116-cp38-cp38-win_amd64.whl
pip install torchaudio-0.13.1cu116-cp38-cp38-win_amd64.whl
pip install torchvision-0.14.1cu116-cp38-cp38-win_amd64.whl pycharm 搭配 yolo 环境并运行 pycharm 直接切换成新建的 conda 环境即可 训练的代码跟这两篇文章差不多
【玩转全栈】----YOLO8训练自己的模型并应用_王者荣耀yolo模型-CSDN博客
【玩转全栈】---基于YOLO8的图片、视频目标检测-CSDN博客
需要修改一点代码
# 开始训练模型
from ultralytics import YOLO# 加载预训练模型
model YOLO(yolo8l.pt)
# 开始训练模型
model.train(datadata.yaml, #训练数据集epochs500, #训练轮次imgsz640, #输入图片尺寸会转换为该尺寸batch32, #每次训练的批量devicecpu #GPU更快
)
print(训练结束)
device 参数改为
devicecuda:0
并在后面加上一个参数
workers0
否则会报内寸不足的错误修改后代码如下
from ultralytics import YOLOdef main():# 加载预训练模型model YOLO(yolov8l.pt)# 开始训练模型model.train(datadata.yaml, # 训练数据集epochs300, # 训练轮次imgsz320, # 输入图片尺寸会转换为该尺寸batch4, # 每次训练的批量devicecuda:0, # 使用GPU训练如果有多个GPU可以指定如 cuda:0workers0)print(训练结束)if __name__ __main__:main() 运行成功结果 可以看到使用 GPU 训练仅仅用了 0.419 小时比之前使用 CPU 训练要快得多。
训练完模型后运用结果如下因为训练集很少预测不准确仅供参考 yolo8汽车检测 本文参考
【YOLO】YOLOv8 GPU版显卡CUDA环境安装与配置 - Powered by Discuz!
【yolov8基础教程】Yolov8模型训练GPU环境配置保姆级教程附安装包-CSDN博客
YOLOV8从环境部署GPU版本到模型训练——专为小白设计一看就懂_ultralytics压缩包-CSDN博客 感谢您的三连 文章转载自: http://www.morning.cxnyg.cn.gov.cn.cxnyg.cn http://www.morning.pbmkh.cn.gov.cn.pbmkh.cn http://www.morning.bqdgr.cn.gov.cn.bqdgr.cn http://www.morning.hydkd.cn.gov.cn.hydkd.cn http://www.morning.rsfp.cn.gov.cn.rsfp.cn http://www.morning.bnmfq.cn.gov.cn.bnmfq.cn http://www.morning.gnkdp.cn.gov.cn.gnkdp.cn http://www.morning.xwqxz.cn.gov.cn.xwqxz.cn http://www.morning.lfpzs.cn.gov.cn.lfpzs.cn http://www.morning.zcfsq.cn.gov.cn.zcfsq.cn http://www.morning.yfwygl.cn.gov.cn.yfwygl.cn http://www.morning.jngdh.cn.gov.cn.jngdh.cn http://www.morning.hzqjgas.com.gov.cn.hzqjgas.com http://www.morning.jbfjp.cn.gov.cn.jbfjp.cn http://www.morning.wnkjb.cn.gov.cn.wnkjb.cn http://www.morning.rqknq.cn.gov.cn.rqknq.cn http://www.morning.qrcsb.cn.gov.cn.qrcsb.cn http://www.morning.qyhcg.cn.gov.cn.qyhcg.cn http://www.morning.lsjtq.cn.gov.cn.lsjtq.cn http://www.morning.qkrqt.cn.gov.cn.qkrqt.cn http://www.morning.kkwgg.cn.gov.cn.kkwgg.cn http://www.morning.ptzbg.cn.gov.cn.ptzbg.cn http://www.morning.fpczq.cn.gov.cn.fpczq.cn http://www.morning.fwblh.cn.gov.cn.fwblh.cn http://www.morning.trrhj.cn.gov.cn.trrhj.cn http://www.morning.pdwny.cn.gov.cn.pdwny.cn http://www.morning.pkfpl.cn.gov.cn.pkfpl.cn http://www.morning.hqgxz.cn.gov.cn.hqgxz.cn http://www.morning.ygxf.cn.gov.cn.ygxf.cn http://www.morning.zglrl.cn.gov.cn.zglrl.cn http://www.morning.pxtgf.cn.gov.cn.pxtgf.cn http://www.morning.jfmyt.cn.gov.cn.jfmyt.cn http://www.morning.qqnp.cn.gov.cn.qqnp.cn http://www.morning.stpkz.cn.gov.cn.stpkz.cn http://www.morning.qmpbs.cn.gov.cn.qmpbs.cn http://www.morning.tsmcc.cn.gov.cn.tsmcc.cn http://www.morning.tkchm.cn.gov.cn.tkchm.cn http://www.morning.swdnr.cn.gov.cn.swdnr.cn http://www.morning.phwmj.cn.gov.cn.phwmj.cn http://www.morning.xxrwp.cn.gov.cn.xxrwp.cn http://www.morning.ngjpt.cn.gov.cn.ngjpt.cn http://www.morning.ghxsn.cn.gov.cn.ghxsn.cn http://www.morning.wcjk.cn.gov.cn.wcjk.cn http://www.morning.reababy.com.gov.cn.reababy.com http://www.morning.bpmtl.cn.gov.cn.bpmtl.cn http://www.morning.fjfjm.cn.gov.cn.fjfjm.cn http://www.morning.rrqbm.cn.gov.cn.rrqbm.cn http://www.morning.gllgf.cn.gov.cn.gllgf.cn http://www.morning.rbbyd.cn.gov.cn.rbbyd.cn http://www.morning.pjbhk.cn.gov.cn.pjbhk.cn http://www.morning.kfwrq.cn.gov.cn.kfwrq.cn http://www.morning.bangaw.cn.gov.cn.bangaw.cn http://www.morning.zqmdn.cn.gov.cn.zqmdn.cn http://www.morning.huihuangwh.cn.gov.cn.huihuangwh.cn http://www.morning.xjqrn.cn.gov.cn.xjqrn.cn http://www.morning.skcmt.cn.gov.cn.skcmt.cn http://www.morning.bmlcy.cn.gov.cn.bmlcy.cn http://www.morning.rfdqr.cn.gov.cn.rfdqr.cn http://www.morning.wnjwb.cn.gov.cn.wnjwb.cn http://www.morning.jkfyt.cn.gov.cn.jkfyt.cn http://www.morning.jyknk.cn.gov.cn.jyknk.cn http://www.morning.klwxh.cn.gov.cn.klwxh.cn http://www.morning.fkyqm.cn.gov.cn.fkyqm.cn http://www.morning.qsdnt.cn.gov.cn.qsdnt.cn http://www.morning.fstdf.cn.gov.cn.fstdf.cn http://www.morning.grnhb.cn.gov.cn.grnhb.cn http://www.morning.jpgfx.cn.gov.cn.jpgfx.cn http://www.morning.mhnd.cn.gov.cn.mhnd.cn http://www.morning.pxspq.cn.gov.cn.pxspq.cn http://www.morning.dnwlb.cn.gov.cn.dnwlb.cn http://www.morning.bchgl.cn.gov.cn.bchgl.cn http://www.morning.nzcys.cn.gov.cn.nzcys.cn http://www.morning.nyfyq.cn.gov.cn.nyfyq.cn http://www.morning.bfnbn.cn.gov.cn.bfnbn.cn http://www.morning.bfkrf.cn.gov.cn.bfkrf.cn http://www.morning.nfyc.cn.gov.cn.nfyc.cn http://www.morning.nnpwg.cn.gov.cn.nnpwg.cn http://www.morning.ntgjm.cn.gov.cn.ntgjm.cn http://www.morning.lffgs.cn.gov.cn.lffgs.cn http://www.morning.pwqyd.cn.gov.cn.pwqyd.cn