利用自己的数据集构建一个简单的tensorflow模型

上次我们谈到了利用爬虫来爬取我们训练用的图片图片数据集，那这次我们就要利用爬到的图片构建一个简单的模型来训练。废话不多说，下面开始。

首先最重要的一点，版本匹配，以下是我用的tensoflow和python版本

tensorflow-cpu——2.7

python——3.7

这里我用的是cpu版本的，因为我的cudn是11.6，cudnn是8.1.0，版本问题比较多，所以没有用gpu版本的。大家在搭建的时候一定要注意自己的驱动环境。如果不知道自己驱动环境，N卡就在英伟达控制面板里查。

 左下角的系统信息里，点进去选择组件，就能查看。

 然后cudnn版本的查看就搜索system文件夹，找到cudnn_version.h，选择记事本打开或者编译器，找到这一行就能知道。我的就是8.1.0。

 关于版本上，如果大家还没有搭建环境，推荐大家安装11.2版本的cudn和8.1版本的cudnn。大家如果找不到8.1的cudnn版本，可