首先声明这个是我自己用于学习的总结文档,同时也感谢以下引用博文的博主“慷慨解囊”,他们将自己的经验分享到社区为我们提供了帮助。文章中如有不妥之处也欢迎大家批评指教。另外这是我第一次发csdn所以对平台规则不甚了解,如遇侵权立刻删帖。
okay!让我们进入正文!
一、环境搭建部分
我采取的方案,一共需要安装五个环境
- Anaconda(我们需要的是这个IDE中的python库)
- Cuda
- Cudnn
- Tensorflow-gpu-x.x.x
需要注意的是:tensorflow不同版本对应的另外三个环境版本是有严格要求的 (如果版本不对应就会出现一系列问题,因此如果后期调试程序的时候出现报错,不妨检查一下自己的版本是否对应)
因此,我将tensorflow的对应版本对照表列出:请参考以下博客内容
https://blog.csdn.net/weixin_44560088/article/details/117457619?ops_request_misc=&request_id=&biz_id=102&utm_term=tensorflow-gpu2.5.0%E5%AF%B9%E5%BA%94cuda&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-.first_rank_v2_pc_rank_v29&spm=1018.2226.3001.4187
接下来是以上环境的安装教程,再次强调:如果按照上述步骤安装之后训练程序过程中报错,首先应该检查的是五个环境的版本是否匹配。
https://blog.csdn.net/weixin_44170512/article/details/103990592
如果不匹配,则可搜索关键词“某某某+ 降级到什么版本”或者“某某某+ 升级到什么版本”(当然,删除重装也是一种不错的选择)
除这些之外,我们还需要手动安装cv2库,csdn上搜索“如何安装cv2库即可”
二、训练部分
接下来我们进入正文、同时也是最有意思的部分。跟着这个视频来做即可:
https://www.bilibili.com/video/BV1r5411t7Db
good luck!
|