为了在Windows环境下成功运行YOLOv3模型,需要正确配置GPU支持。具体的配置方法可以在这里找到详细的指南。
本项目使用了qqwweee在GitHub上开源的keras-yolo3代码库,项目地址为:https://github.com/qqwweee/keras-yolo3。
我的实际配置情况如下:
- 操作系统:Windows 10
- TensorBoard 版本:1.8.0
- TensorFlow-GPU 版本:1.8.0
- 安装命令:pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.8.0
- Keras 版本:2.2.4
- 安装命令:pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ keras==2.2.4
- Python 版本:3.6.8
- CUDA 版本:9.0
- CUDA 下载地址:NVIDIA CUDA 9.0
通过上述配置,能够确保YOLOv3模型在Windows 10系统上的顺利运行,同时利用GPU加速模型训练过程,提高训练效率。