适合0基础的超级详细YOLO11的GPU环境配置与运行(适用于YOLOv8)

写于前面的一句话:

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

要配置yolov7环境使用gpu,可以按照以下步骤进行操作: 1. 首先确保你已经安装了合适的显卡驱动程序,并且你的显卡支持CUDA。 2. 安装CUDA Toolkit。可以在NVIDIA官方网站上下载适合你系统的CUDA Toolkit,并按照官方指南进行安装。 3. 安装cuDNN。cuDNN是一个专门用于深度神经网络的加速库,也需要从NVIDIA官方网站上下载相应版本,并按照官方指南进行安装。 4. 创建一个新的conda虚拟环境。使用以下命令创建一个名为yolov7的虚拟环境,并指定Python版本为3.7: ``` conda create -n yolov7 python=3.7 ``` 5. 激活新创建的虚拟环境。使用以下命令激活yolov7环境: ``` conda activate yolov7 ``` 6. 安装必要的依赖库。使用以下命令安装yolov7所需的依赖库: ``` pip install numpy opencv-python tensorflow-gpu ``` 7. 克隆yolov7的代码仓库。使用以下命令克隆yolov7的代码仓库: ``` git clone https://github.com/ultralytics/yolov3.git ``` 8. 进入yolov7代码仓库的目录,并下载预训练的权重文件。使用以下命令进入目录: ``` cd yolov3 ``` 然后使用以下命令下载预训练的权重文件: ``` wget https://pjreddie.com/media/files/yolov3.weights ``` 9. 将预训练的权重文件转换为TensorFlow格式。使用以下命令将权重文件转换为TensorFlow格式: ``` python convert.py --weights yolov3.weights --output ./checkpoints/yolov3.tf ``` 这将会生成一个名为yolov3.tf的文件。 10. 环境配置完成。现在你可以在配置好的yolov7环境运行相关程序了。
评论 22
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值