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

这篇具有很好参考价值的文章主要介绍了在pycharm中配置GPU训练环境(Anaconda)(yolov5)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

目录

1. 具体的配置过程:

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

3. conda常用命令:

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

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

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

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

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

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

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


1. 具体的配置过程:

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

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

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

conda info --envs

在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5
2:在指定目录下创建新的虚拟环境,输入命令:

conda create --prefix=C:/ProgramData/Anaconda3/envs/pytorch python=3.8

        其中C:/ProgramData/Anaconda3/envs 是创建的目录所在位置;/pytorch是所创建的环境的名称 python=3.8是创建的python的版本。
在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5

        可以看到,验证确实创建在我们想要的位置
在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5
        OK啦,创建成功:
在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5
在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5

3. conda常用命令:

 常用命令大全

  • 删除虚拟环境的代码如下(yolo5表示env名字) 
conda env remove -n yolo5
  • 激活虚拟环境
conda activate F:\Anaconda\envs\yolo5

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

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

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

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

    在pycharm中配置GPU训练环境(Anaconda)(yolov5),pycharm,python,yolov5
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分钟,不得不说真香)

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

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

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

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

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

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

该问题出现的原因在于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的区别:

虚拟环境中cuda版本与系统cuda版本不同有什么影响

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

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

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

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

大家如果有开源的鬼灭之刃数据集,可以考虑分享给bz一下,嘿嘿;有问题欢迎在评论区留言!文章来源地址https://www.toymoban.com/news/detail-740387.html

到了这里,关于在pycharm中配置GPU训练环境(Anaconda)(yolov5)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 【零基础玩转yolov5】yolov5训练自己的数据集(CPU训练+GPU训练)

     博主也是最近开始玩yolov5的,甚至也是最近开始使用python的,很多东西都没有接触过,因此训练自己的数据集花了不少时间,所以想写篇博客记录一下,希望同样是零基础的小伙伴们可以更加轻松的上手。同时大家如果发现了错误和理解偏差,欢迎指正。 参考资料: Yolo

    2024年02月06日
    浏览(52)
  • 配置Cuda及Cudnn,在Anaconda创建虚拟环境,安装GPU版Pytorch,并在Jupyter noterbook及Pycharm中调用【极其详细】

    目录  一、配置Cuda及Cudnn (一)下载Cuda 1、查看电脑自带的Cuda版本 2、下载相应版本Cuda安装包 3、安装Cuda 4、配置变量 (二)下载Cudnn 二、创建虚拟环境 三、虚拟环境中安装GPU版Pytorch (一)有关环境的基本指令 (二)安装GPU版Pytorch 四、将虚拟环境在Jupyter noterbook及Pycha

    2024年04月28日
    浏览(35)
  • YOLOv5训练速度慢 GPU占用率低

       用Yolo5跑自己的数据集,torch_GPU和CUDA都安装了,使用GPU跑模型,GPU的占用率为1%,设置的epoch为300,batch size为32,大概 6min 多/epoch,数据集很小,刚过一千。在网上查了相关的文章,发现可能是读取数据造成的,把cache-image的action=\\\'store_true\\\'改成\\\'store_false\\\'。  改成 修改后 每

    2024年02月13日
    浏览(72)
  • 【基于Ubuntu下Yolov5的目标识别】保姆级教程 | 虚拟机安装 - Ubuntu安装 - 环境配置(Anaconda/Pytorch/Vscode/Yolov5) |全过程图文by.Akaxi

    目录 一.【YOLOV5算法原理】 1.输入端 2.Backbone 3.Neck 4.输出端 二.【系统环境】 1.虚拟机的安装与创建 2.安装Ubuntu操作系统 3.环境的配置 3.1.Ubuntu下Anacoda安装以及虚拟环境配置 3.2.Pytorch安装 3.3.Vscode安装 3.4.Yolov5源码及环境获取安装 三. 【测试Yolov5】 四. 【实现自己输入图片/笔

    2024年02月05日
    浏览(53)
  • 在pycharm中用anaconda配置python环境

    想在pycharm中调用某个python环境,首先要有这个python文件,因此本文将先介绍如何用anaconda下载不同版本的python编译器,再介绍如何用pycharm添加代码文件需要的python编译器。 其中有人问:下什么python编译器,直接在pycharm在选择编译器那里,选择new environment不就完了,下载+调用

    2024年02月15日
    浏览(30)
  • yolov5的运行环境配置、参数修改和训练命令,并将训练的pt模型转换为rknn模型,并推理

    目录 一.yolov5的运行环境配置(Windows系统) 1、NVIDIA驱动安装与更新  2、Anaconda 的安装 3、Pytorch环境安装  4、pycharm安装 二.yolov5训练数据集,得到best.pt(Windows系统)  1.下载YOLO项目代码  2.环境安装 3. 数据集下载: 三.best.pt转为onnx模型(Windows系统) 四.best.onnx转为best.rknn模型(Linu

    2024年02月13日
    浏览(32)
  • 技巧篇:Mac 环境PyCharm 配置 python Anaconda

    在 python 开发中我们最常用的IDE就是PyCharm,有关PyCharm的优点这里就不在赘述。在项目开发中我们经常用到许多第三方库,用的最多的命令就是pip install 第三方库名 进行安装。现在你可以使用一个工具来帮你解决经常安装第三方库的麻烦,这个工具就是Anaconda。 Anaconda集成了

    2024年01月23日
    浏览(33)
  • pycharm配置anaconda环境时找不到python.exe

    问题描述 : 在一台新电脑上配置anaconda环境时,发现pycharm在设置解释器时,在conda环境中找不到anaconda已经创建好的python解释器可执行文件,其显示如下:  只显示创建的虚拟环境中的那些文件夹: 搜了半天总结出问题:pycharm版本过新。 解决方法: 在选择环境时,选择系统

    2024年02月08日
    浏览(51)
  • Anaconda + Pycharm,利用Anaconda安装python并配置虚拟环境,包括sklearn、pytorch的安装

    前言:在安装前,如果自己的电脑名称里带中文,请先把自己的电脑重命名,切记不要带中文,否则可能会出错。 说明:本人用这套是为了做深度学习,如果我们一样,可以看下去,不一样也有参考价值。 本文不包括pycharm的安装教程! Anaconda主要是方便后续导包,装完Ana

    2024年02月05日
    浏览(54)
  • AutoDL+pycharm+yolov5远程租服务器训练数据集

    自己电脑算力简直太差劲了,在AUTOdl上租个0.88/h的服务器跑yolov5代码,找了好多教程,鼓弄了两天时间我的代码才算能跑起来,感觉市面上的博客写的都不够详细,包括我遇到的一些问题,我整理了一下发出来也供自己参考 目录 1.AutoDL租用服务器 2.下载Pycharm远程开发 2.1 下载

    2024年02月04日
    浏览(39)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包