## 安装
### 我的GPU信息
环境要先安装显卡驱动和CUDA
Driver Version: 531.41 CUDA Version: 12.1
### tensorflow安装
```
参考： https://tensorflow.google.cn/install?hl=zh-cn
# Requires the latest pip
pip install --upgrade pip
# Current stable release for CPU and GPU
pip install tensorflow-gpu==2.8.1
```
## 训练
```
在train_model.py 修改数据集路基 mycatdog2
model.train(fpath="F:\数据集\mycatdog2")
执行:
python train_model.py
```
第4例 代码基于卷积神经网络VGG的猫狗图像识别
需积分: 5 189 浏览量
2023-06-16
23:23:13
上传
评论 1
收藏 23KB ZIP 举报
VGG_test.zip （20个子文件） 
VGG_test 
train_model.py 5KB output my.log 165B predict_model.py 812B network vgg11.py 3KB vgg16.py 4KB __pycache__ vgg16.cpython-39.pyc 3KB lenet.cpython-39.pyc 1KB alexnet.cpython-39.pyc 2KB vgg16.cpython-38.pyc 3KB vgg16.cpython-310.pyc 3KB README.md 507B data_process data_manager.py 1KB myreaddata.py 1013B __pycache__ myreaddata.cpython-310.pyc 1KB data_manager.cpython-310.pyc 1KB myreaddata.cpython-38.pyc 1KB data_manager.cpython-38.pyc 1KB myreaddata.cpython-39.pyc 1KB data_manager.cpython-39.pyc 1KB check_gpus.py 220B资源评论
