
笔记
文章平均质量分 70
Dcurry
这个作者很懒,什么都没留下…
展开
-
深度学习Tensorflow的坑(二:训练一个epoch卡住不动,程序不终止也不继续)
第一篇承接上文在搭载好环境之后,满心欢喜准备搭载模型的绘图(损失曲线、验证精度)功能,我就先使用10个epoch跑一下,结果。它卡在199/200了!之前从没遇到过这个情况。先看控制台报错,说是导入CUDNN的DLL文件不成功,得,又得复制粘贴了。接着跑,还是不成功。网上很多报错都是pytorch版本得dataloader,不太能够适配。我就开始自身琢磨,我发现训练过程是没问题得,那么问题是出在我的valmodel,最后一步savemodel根本没走过去(QAQ深度学习不会打断点啊!)。那我的原创 2021-11-22 10:54:45 · 14882 阅读 · 0 评论 -
深度学习Tensorflow的坑(一:无法使用GPU加速训练深度学习模型,CUDA与CUDNN相关)
项目场景:本研究僧为了秋招卷大厂放弃了半年的深度学习,重回Python的一些坑记录问题与解决:1:版本问题想用GPU加速深度学习的训练一定不能同时下载tensorflow和tensorflow-gpu。如果遇到了什么tensorflow找不到的bug,请直接删除所有带有tensorflow的包,然后找到对应版本的下载对应版本是指CUDNN和CUDA以及你自身显卡实力对应的tensorflow-gpu版本。2.CUDA和CUDNN下载(2、3一起看食用体验更佳!)CUDA下载比较简单CU原创 2021-11-21 17:00:17 · 4040 阅读 · 0 评论