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