在pycharm中配置GPU训练环境(Anaconda)(yolov5)

目录

[1. 具体的配置过程:](#1. 具体的配置过程:)

[2. 在指定位置(路径)创建虚拟环境:](#2. 在指定位置(路径)创建虚拟环境:)

[3. conda常用命令:](#3. conda常用命令:)

[4: 在跑模型时候遇到的一些问题:](#4: 在跑模型时候遇到的一些问题:)

[4.1: conda添加python解释器找不到对应的python.exe文件](#4.1: conda添加python解释器找不到对应的python.exe文件)

[4.2: 报错"OSError: WinError 1455 页面文件太小,无法完成操作。"](#4.2: 报错“OSError: [WinError 1455] 页面文件太小,无法完成操作。”)

[4.3: 报错"CUDA out of memory. Tried to allocate 14.00 MiB](#4.3: 报错“CUDA out of memory. Tried to allocate 14.00 MiB)

[5: CUDA11.1版本出现NAN的问题原因及解决措施:](#5: CUDA11.1版本出现NAN的问题原因及解决措施:)

[6: 虚拟环境中的cuda和系统中cuda的区别:](#6: 虚拟环境中的cuda和系统中cuda的区别:)

[7: 对于动漫角色,训练效果不是很好 :](#7: 对于动漫角色,训练效果不是很好 :)


1. 具体的配置过程:

手把手教你Anaconda安装虚拟环境配置yolov5

2. 在指定位置(路径)创建虚拟环境:

1:使用命令查看当前拥有的虚拟环境

bash 复制代码
conda info --envs


2:在指定目录下创建新的虚拟环境,输入命令:

bash 复制代码
conda create --prefix=C:/ProgramData/Anaconda3/envs/pytorch python=3.8

其中C:/ProgramData/Anaconda3/envs 是创建的目录所在位置;/pytorch是所创建的环境的名称 python=3.8是创建的python的版本。

可以看到,验证确实创建在我们想要的位置

OK啦,创建成功:

3. conda常用命令:

常用命令大全

  • 删除虚拟环境的代码如下(yolo5表示env名字)

    conda env remove -n yolo5

  • 激活虚拟环境

cpp 复制代码
conda activate F:\Anaconda\envs\yolo5

4: 在跑模型时候遇到的一些问题:

4.1: conda添加python解释器找不到对应的python.exe文件

网上有很多解决方法,我参考了这个:(也是用GPU训练的,如果解释器是anaconda虚拟环境中的python,那效果应该是一样的)

  • 直接在Virtualenv Enviroment中找conda的虚拟环境里的python.exe就可以

  • 直接在System Environment中找conda的虚拟环境就可以

4.2: 报错"OSError: WinError 1455 页面文件太小,无法完成操作。"

这个好像和给cuda虚拟环境所在盘分配的虚拟内存有关:

具体可以参考下边这个博客:

页面文件太小,无法完成操作。

4.3: 报错"CUDA out of memory. Tried to allocate 14.00 MiB

这个应该是训练的模型所设置的batch-size太大了,GPU的显存满足不了

把batch-size调小一些,如设置常用的256 128 64 32 16等,我是把yolov5中的16改为了8,然后问题就迎刃而解了。

最后附上使用GPU训练yolov5模型成功的截图:

(之前用CPU跑同样规模的数据集跑了2个小时,用了GPU只需要16分钟,不得不说真香)

显卡配置:(有钱一定换块好的)

5: CUDA11.1版本出现NAN的问题原因及解决措施:

我在完成上述步骤后,出现了一些错误,具体表现为如下形式:

该问题出现的原因在于CUDA版本较高(11.1),最好采用10.2的CUDA版本 ,只需要在虚拟环境中安装10.2的cuda即可,不需要在系统中重新安装。

解决方法可以参考下述博客:

YOLOv5s GTX 1660 Ti训练时出现,box,obj,cla全是nan的问题P、R、mAP都是0,Pytorch和cuda、cudnn版本不对

GTX 16XX系显卡 yolov5训练结果出现NAN的问题

yolov5 训练自己的数据集后,检测不到目标的解决办法

此外,也可能是如下问题导致的:

服务器上训练好的yolov5数据集在自己电脑上什么都检测不出来(已解决)

6: 虚拟环境中的cuda和系统中cuda的区别:

https://blog.csdn.net/qq_42744739/article/details/118640226

7: 对于动漫角色,训练效果不是很好 :

原本想训练一个自动识别视频中动漫角色(九柱)的模型,但是拿 恋柱------甘露寺蜜璃和水柱------富冈义勇做了小实验(每个选取了各40张图片,其中3张作为验证集),在batch_size=8的情况下epoch=200,最后只能实现识别这两个人物不会出太大问题,但是你喂给它一张猫或狗或者其他柱可能会识别错误(可能是因为数据集太小的缘故,同时动漫人物本身识别也较为困难)

大家如果有开源的鬼灭之刃数据集,可以考虑分享给bz一下,嘿嘿;有问题欢迎在评论区留言!

相关推荐
代码山河7 小时前
Python就业方向全解析:后端、爬虫、数据分析、AI你选哪个?
python·学习·架构·教程·面向对象·项目
在世修行7 小时前
智学文枢:题目流水线
python·多线程·状态机·pyside
梯度下降者7 小时前
零基础入门Python 爬虫
爬虫·python·数据分析·入门指南·webcrawler
“AI国潮设计-小江”9 小时前
【AIGC实战】Python+Stable Diffusion打造“英歌舞财神”IP,解锁“马到成功”等爆款情绪营销文案设计
开发语言·人工智能·python·prompt·aigc
benchmark_cc10 小时前
用 Python 拉 A 股全市场实时快照:Notebook 请求、结果校验与易错点
python·数据分析·pandas·量化交易·股票数据·quantdash
伞伞悦读10 小时前
【第70期】Adam二阶矩不是方差:偏差校正、动量与AdamW完整实验
python
外收内放10 小时前
Python基础语法练习题(73)
java·前端·python
锋行天下11 小时前
Langgraph HITL 中断运行,用户输入后输出打字效果
人工智能·python
时早玖沐11 小时前
给 AstrBot 写一个天气推送插件:12 次提交、6 个坑
python·开源·ai编程
触底反弹11 小时前
🚀 FastAPI 五大重难点深度剖析:从 500 报错到彻底搞懂 Depends、Pydantic 与 SQLAlchemy
python·fastapi