Yolo v5训练自己的数据集,新手小白废话连篇,写给自己

研一新生,记性不好,仅以此来记录平平无奇的研一生活,只要不学习就还算有趣,每一天都在膜拜大神。

导师布置完让我用pytorch啥啥训练我们自己标注的数据集,我就在摸爬滚打中前进前进,还没开学,所以走了很多弯路,后来我就提前去学校了,然后不会的就可以不要脸的去问你的同门!然后你就会得到同门大神的指点哈哈哈。

正片开始

提示:一开始我找了很多教程,然后发现了一个非常详细的教程,不需要下载CUDA和cudnn的一个,教程在这里,但是我们老师说不行(好像说是训练的时候不稳定),要单独下载CUDA和cudnn,所以我又兜兜转转下载了这两个。

 一.准备

1.买电脑:显卡是英伟达的,直接买个带劲儿的游戏本,或者我有个同学买的轻薄本,让老师给配一个机箱,用向日葵这个软件然后调用也可以,各有各的好处,不过我本人不打游戏,真的很讨厌游戏本,迫不得已买了游戏本好难过。

2.喜提新电脑回家之后,打开英伟达控制面板(桌面右键)

扫描二维码关注公众号,回复: 15962883 查看本文章

  帮助-----系统信息-----组件

看那个第三行,CUDA11.4啥的,嗷嗷顶配

有的显卡配置比较低可以去官网更新一下显卡驱动,点这里

 设备管理器,显示适配器看你自己显卡型号,GeForce -----3060 Laptop,下载就行(这是我的)

二.下载各个大软件

1.Acanconda

安装了这个以后我们就不在cmd里搞了,就在这里下各种环境的包啥的,网址点这里

这是我下的,然后安装就行, 完了他跳让你选,选just me,只有我这个当前用户使用。---next,安装路径很重要,放在D盘,专门给他搞个文件夹,因为所有的环境都在这里

 搞好了下一步,这个

我搞完了,这是别人的图片,我借来用用(侵删) ,红了也打勾,把anac搞到环境变量里去,就不用手动搞了。

 完了就完事了这个,Anaconda Prompt(Anaconda)只用这个

2.pycharm

这个很简单,随便安安,点这里进官网,下community的就行,安装在d盘,默认安装就行

3.CUDA

有啥不会找百度,百度搜索cuda官方下载,或者点这里,我刚刚试了突然打不开了,可能是学校校园网太次,打开之后选就行了,别纠结,就11.0就行,因为那个支持最高是11.2的好像是(2021.09)

如图,自定义安装,这是别人的图,跟他不一样,除了Visual Studio那个,其余的我都选了

 

 同样,搞到d盘去,安装就行。

这里有个环境变量的问题,这个问题我没遇到,他自己都搞好了,不用手动添加,有的博主说需要,右键-----属性------高级系统设置------环境变量-----双击path----把CUDA应该是4个添加进去就行。

验证是否成功:打开anac,输入nvcc -V

 成功了,欧耶

4.cudnn

这个没难度,点这里,放心注册,然后下载

找到支持你已安装的CUDA版本的CUDNN,下载就可以。解压下载的压缩包,并将里边的bin文件夹,include文件夹,lib文件夹复制到CUDA的安装目录里面。(注:之前CUDA安装目录里面也有这些文件夹,要把CUDA的这三个文件给覆盖掉)

就可以了!

5.pytorch环境

环境这个东西就是,你安,然后你自己觉得,啊,终于安完了,跑代码的时候会发现还有环境没安,少环境,然后你再接着安,我吐 

anaconda(简称anac),打开他,输入conda list

新安装的只有一个base主环境,解释一下就是,你买了一个100平方的房子base,你需要给他划分出卧室,厨房,书房,厕所。

所以我们创建一个屋子给pytorch

输入conda create -n pytorch python=3.7

(pytorch是给这个房间起的名字,你也可以起别的,我的是pt17,然后后面那个是版本号)

回车,遇到问你y/n,就打个y,回车

然后就创建好了,再次输入conda list ,就能看到你建的那个装pytorch的屋子了

然后就激活他,输入conda activate pytorch
激活之后就在哪里下载各种的烦人的环境。

好,最艰难的地方来了!pytorch的下载!我吐了!官网下载不来人之常情,因为外网嘛,可以理解,为啥清华源人家用的都好用,我用就不行?好吧,不抱怨了,反正就三个常规办法

(1)直接外网下

激活你自己的pytorch环境之后,复制这个

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge 

回车,可能会成功,可能会成仁,成功不必说,成仁会出现以下

 这种就是失败了,直接关掉重来,大概率因为网不好或者其他我不知道的原因,建议晚上试,我同门下载成功,我并没有成功。

同样可以用pip下载:命令都是直接在官网搞得,没有技术含量,截图很费劲,不截图了,直接搞,进入pytorch环境,输入以下,回车

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

马上就完事了,就差一点点,然后就是很难过的红红红,如下

错误大约就是time out,家里的网络太差(我觉得网还行),反正也没成功。

建议这时直接翻墙,应该会成功,我没搞。

(2)镜像

点这个,找到你所喜爱的版本哈哈哈,下载下来,比如

 然后找1.8.0对应的torchvision版本是多少(官网找)

 下完放在  Anaconda/envs/pytorch/里面

然后直接d:进入到d盘,然后用cd指令进入code(cd /code),然后进入anac,一直进,最后进入pytorch,输入pip install py摁tab自动补齐安装包名字,回车下载。这个我也没成功,我不知道为啥,哭了。

(3)安装包直接复制

(2)没成功,后来同门说将下载的pytorch里的lib文件夹里的site-packages的三个包直接复制到D:\Anaconda\envs\ptg\Lib\site-packages里就会成功,我没试,待考察。

(4)开挂法

没错,我就是用了这个方法,同门下载好了直接拷给我的,对不起哈哈哈,我是个失败者,不过直接得到就是轻松且好用!如果上述的方法你也没有成功且你没有一个像我这么好的同门的话,淘宝万能。虽然我没成功,但上述的方法我都亲身经历且试过,唬人够用了。

至此,你的大环境就差不多安好了,还有一些小一点的环境,跑的时候缺啥安啥。

以此开启我的研究生生涯,小废物一个,说错的地方欢迎指正。

猜你喜欢

转载自blog.csdn.net/Zosse/article/details/120541888
今日推荐