适合0基础的超级详细YOLO11的GPU环境配置与运行(适用于ultralytics项目包里的所有YOLO版本)

写于前面的一句话:

Pycharm+Anaconda+Cuda

本博客内容面向Windows操作系统(C盘保留20G内存以上),电脑需要有显卡和科学上网工具 

怎么看自己电脑有没有显卡:右键此电脑(我的电脑)--->管理

10c31bc1d158462988e040a8601f4b32.png

8b6c855293f344f58f1492964346945c.png

1. 在D盘(不建议C盘) 创建三个文件夹(Anaconda3、pycharm、yolo11)

662dafd50e5a480b9bbbb837bf332dad.png

2.下载Anaconda(打开科学上网工具

网址:https://www.anaconda.com/download/success

点击Download开始下载

f078776618574f6387019aec969951fd.png

下载完成后运行:

cb664f9c023a423c85a5e44398fb87d9.png

7949bde65c524b9d96a9e8bb8593592b.png

6148b45f7c4c453b9a8c8c09757bfba3.png

选择刚刚创建文件夹的位置

7eeedc7bd4a247eab00ad3f7c86620b5.png

选择这三个:

60c3ae64170d4157821f029e7a344b2f.png

安装完成(这一步时间比较久,耐心等待):

95b29b85d39943779cfb4a5c8c94b825.png

97ae0d05745d4ad88428066830ad8279.png

在我们的电脑里配置Anaconda,此电脑(我的电脑)--->属性

c6d6ce32810e454780ef2b988535b405.png

找到高级系统设置:

b5c85fe053f44919b361667ada97d18e.png

384d4a0cc91042cc9f40c73687cf1721.png

dd6dbf3714a74dd2a2fd98fab679c185.png

新建:

c754df5b81ff480cb0c85ce057f95c5e.png

打开你的Anaconda安装文件夹,新建这四个地址:

1373edb668144cb2aeb273de1eebcb8e.png

然后确定,再确定

1bdcb1f153684dbea3599b7f68ddc7c2.png

3 pycharm安装打开科学上网工具

网址:https://www.jetbrains.com/pycharm/download/other.html

往下滑,找到:

38948351c1974a6fa2ae87f0ad98456f.png

点击下载,下载完成有运行安装包

9f9daea291014e089db8576698ce0793.png

找到pycharm文件夹:

5452a62079a4455393fc96307a7bdbfe.png

8c0722a8518648f38a4444c330b9e946.png

36f9ec17e978408d9708bdc087d43529.png

安装完成:

1d790a526b534579b3b239e3b0117818.png

在我们的电脑里配置pycharm,此电脑(我的电脑)--->属性

c6d6ce32810e454780ef2b988535b405.png

找到高级系统设置:

b5c85fe053f44919b361667ada97d18e.png

384d4a0cc91042cc9f40c73687cf1721.png

dd6dbf3714a74dd2a2fd98fab679c185.png

新建:

9771ce0a8e9f4b868cdd28ea6a95abbb.png

找到pycharm文件夹,存放这一个地址:

c00d6eb590cc4e6e8d481d5306d709fd.png

确定,再确定:

1a000b55a1c748b9b50ed68b601fef35.png

4 安装cuda与cudnn(用于GPU训练模型)(打开科学上网工具)

网址:https://developer.nvidia.com/cuda-toolkit-archive

往下滑,找到:

2fa2a4bef2c34f53b70910b26d21ce27.png

Windows10选10,是Windows11的选11,点击下载

8491d07bb2d545848c063ec4b5d0524d.png

下载完成后,点击运行程序:

默认地址,选择ok

4fa44c22c3044bbcaae651a2494fadf2.png

cebbb60e35f04940937d1c88c2db4526.png

b709c9e106cb46fe80bb2b7ffcdaa7cd.png

b8ec7a23e14542a4b406543e4b240619.png

默认地址,下一步:

b700ce874e654a8d864b5a108ccdaa47.png

3b5d84fe48f14bd8973e27e9bd1d520e.png

安装完成:

2f182282342c4edbad1a10497e0491bd.png

da6fbbea1ca946bbb4c980d30a9a41c7.png

 找到高级系统设置:

b5c85fe053f44919b361667ada97d18e.png

384d4a0cc91042cc9f40c73687cf1721.png

看到这就代表安装cuda成功了:

(1):

(2):

安装cudnn:(需要有英伟达的账号,大家要是没有,我上传到百度网盘了,可以下载,对应的是cuda的11.5版本)

通过百度网盘分享的文件:cudnn-windows-x86_64-8.3.2.44_cuda1...
链接:https://pan.baidu.com/s/1u3lmoYEufcRQ3i008bv_Yw 
提取码:puza 

网址:https://developer.nvidia.com/rdp/cudnn-archive

267875b565a14c3cab6510e5a89f9331.png

登录成功后:往下滑

e73f34e8446b4b3a94d3bc69e0547043.png

点击下载

509af44cc56a4b19a5e5dbb19e206685.png

下载完成后解压缩:复制这三个文件:

b5d65a78e5004a75ab4fc0aad3d46673.png

找到cuda安装地址:c盘

11bfd728739143f3818dc92b040bc8f8.png

ad94c522bf8444a28dacd2d20cd8e124.png

复制到这里:

b7e5a73f754a4a768951753e63872082.png

a90012fd63194e988c29aa90e2bc07c8.png

至此yolo11需要用到的pycharm,anaconda,cuda安装完毕

5 yolo11模型训练

在训练yolo11的时候需要数据集,推荐一个网站:https://universe.roboflow.com/

 我随便选取主页的一个数据集下载:

5c514d8b4c98428d84f03397bcc9089d.png

9a7faf84831a4d7792fe7f92ff77b3bb.png

3e202c01918147a595179d3b97fa77ba.png

c1fd546ba8a2488da8e7fe9b4946342d.png

下载完成后解压缩:训练所需要的四个文件

206078ddb0874aaea0bc2618b39f48db.png

下载yolo11代码:https://github.com/ultralytics/ultralytics

df268c7fd5cc42118bb94a518bf86063.png

下载完成后放到创建的yolo11文件夹里解压缩:

创建一个datasets文件夹,放入刚刚下载的数据集

0faaf8d7ad694ea980180cb71badf8f7.png

715f319d59224426a52d917b9005a59e.png

ultralytics-main右键pycharm打开:

661712bc99e64911958136b1dac8dfe4.png

b1d99d9d2b6b47f58c79dc0862394ecb.png

00d968e28feb431ba66826e9a3533b9a.png

新建一个train

74ce60bcbd784ca6ab9160e694805dac.png

5261ceba4e364887b9d865e9c2bb21de.png

train.py里的这两个路径为绝对路径:(根据自己的电脑的绝对路径修改

26356c42b6db4b3b9a50a1fd6de9091a.png

train.py代码:

import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(r'D:\software\yolo11\ultralytics-main\ultralytics\cfg\models\11\yolo11.yaml')
    model.train(data=r'D:\software\yolo11\ultralytics-main\datasets\data.yaml',
                cache=False,
                imgsz=640,
                epochs=200,
                single_cls=False,  # 是否是单类别检测
                batch=16,
                close_mosaic=10,
                workers=0,
                device='0',
                optimizer='SGD',
                amp=True,
                project='runs/train',
                name='exp',
                )



修改datasets里data.yaml的路径:(绝对路径)(根据自己的电脑的绝对路径修改

2196828780894cfbab0bfa040760bd9f.png

到此数据集和训练代码操作完毕,开始创建yolo11的anaconda环境:

搜索里输入cmd,然后回车:(开启科学上网工具

6b531c6175df4e039fd74dc708f04389.png

ee3cfbb32ba7480fa43e7f88af63c658.png

输入:conda create -n yolo11 python=3.10   然后回车

07d2df9412634d96933056274a3f4ac5.png

再回车:

d1f054e19d2a4843b3291d141674ecf1.png

激活yolo11: activate yolo11

b2d8871d417b422798e42224d2c4ab9a.png

输入:pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117      回车(一定开启科学上网工具)

a3822002b0bc4f0980644816e53fb488.png

1d15fe549a894653839eb982b30ccd1f.png

安装完毕:

0163229dd0034ccba6ddd25e6ff344d2.png

再次输入:pip install -U ultralytics    回车

53d386db60794596be681c7003b1f15d.png

安装完毕:

e50ac7a82cfd45b69251b8724fc0cdbf.png

到此yolo11所需环境安装完毕!!!

在pycharm里调用刚刚创建的yolo11环境:

 右下角:a19360d2ea144c5f9e8c3448073d4391.png

ca405fe88f5649ff9c592d6183b77597.png

5834988aac9d4a9e8f561e2ac2fb748c.png

f8e1fa8d2948478a9c0049f9a4161a74.png

等待pycharm加载一会所需要的库

加载完成后,右键运行train.py

a6596282c4694fe09497fb6268a06218.png

成功打印模型结构:

80d1704bca344fa7b2c8342945e1d014.png

开始模型训练:

8108f28515284e30beea5461079a2fcd.png

到此全文结束!!!

到此本文的正式分享内容就结束了,在这里给大家推荐我的YOLOv11改进有效涨点专栏,本专栏目前为新开的,后期我会根据各种前沿顶会进行论文复现,也会对一些老的改进机制进行补充,如果大家觉得本文帮助到你了,订阅本专栏,关注后续更多的更新~

YOLOv11有效涨点专栏

a06d4733d6d04c91b1f35aec23a38b9a.png

### 关于 YOLO v11 运行代码的相关信息 YOLO v11 是一种强大的目标检测框架,支持多种任务,包括但不限于目标检测、实例分割以及多目标跟踪。为了运行 YOLO v11 代码,通常需要完成以下几个方面的配置和操作。 #### 1. 数据集准备 在运行 YOLO v11 训练之前,需准备好数据集并按照特定格式组织。以下是常见的数据集结构: ```plaintext dataset/ ├── images/ │ ├── train/ │ └── val/ └── labels/ ├── train/ └── val/ ``` 其中 `images` 文件夹用于存储图像文件,而 `labels` 文件夹则保存对应的标注文件(通常是 `.txt` 格式的边界框坐标)。对于实例分割任务,可能还需要额外的点集合标注文件[^2]。 #### 2. 配置 YAML 文件 YOLO v11 要求定义一个 YAML 文件来描述数据集路径和其他参数。以下是一个典型的 `data_cls.yaml` 文件示例: ```yaml # dataset path train: ./images/train val: ./images/test test: ./images/test # number of classes nc: 1 # class names names: ['object'] ``` 此文件指定了训练集、验证集的位置,类别数量 (`nc`) 和类别的名称列表 (`names`) [^3]。 #### 3. 编写训练脚本 编写 Python 脚本来加载模型并启动训练过程。下面是一份基于 Ultralytics 库的简单训练脚本示例: ```python from ultralytics.models import YOLO import os os.environ['KMP_DUPLICATE_LIB_OK'] = 'True' if __name__ == '__main__': # 加载预定义的 YOLOv11 模型架构 model = YOLO(model='ultralytics/cfg/models/11/yolo11.yaml') # 开始训练 model.train( data='./data_cls.yaml', # 数据集配置文件路径 epochs=500, # 总共训练轮数 batch=48, # 批次大小 device='0', # GPU 设备编号 (如果使用 CPU,则设为 'cpu') imgsz=640, # 输入图片尺寸 workers=2, # 工作线程数 cache=False, # 是否缓存数据到内存 amp=True, # 是否启用自动混合精度 mosaic=False, # 是否启用马赛克增强 project='runs/train', # 输出目录前缀 name='exp' # 实验名称 ) ``` 这段代码展示了如何初始化 YOLO 模型并通过指定的数据集进行训练。注意调整超参数以适应具体应用场景[^3]。 #### 4. 启动训练命令 确保所有依赖项已正确安装后,在终端执行如下命令即可开始训练: ```bash python train_cls.py ``` 这会调用前面编写的 `train_cls.py` 脚本,并依据设定好的参数逐步优化模型性能[^3]。 --- ###
评论 22
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值