tensorflow-gpu Win10 环境搭建
上篇文章讲的手写数字识别,识别率达到99.25%。笔者想做个手写汉字识别,奈何Mac 本配置局限性,本次使用Win10系统开发。笔记本配置 16G内存,256G固态硬盘,1T机械硬盘,显卡GeForce GTX 960M(稍有不足)闲话少说,开始进入正题。
1、安装Python
- python 官网地址:https://www.python.org/
笔者开始电脑安装好了python,python版本是3.7x。然后安装通过 pip install tensorflow-gpu,总提示找不到。后来去了https://pypi.org/project/tensorflow-gpu/1.10.0/ 发现目前新版本只支持python 3.6x。安装时选择python 选择3.6X版本
2、安装tensorflow-gpu
pip install --upgrade tensorflow-gpu
安装 tensorflow-gpu 完成。
3、安装CUDA和cuDNN
- CUDA 9.0
- 下载地址:https://developer.nvidia.com/cuda-downloads
- 目前不支持CUDA9.2,不要下载。
- 下载后直接安装即可
- cuDNN v7.1.3
- 下载地址:https://developer.nvidia.com/rdp/cudnn-archive
- 下载后解压,放到文件夹下,配置系统环境变量。
- 注意:高版本在使用时会报 kernel reported driver version not implemented on Windows
安装完成后启动上次编写好的手写数字识别
- 安装 Image
pip install Image
- 安装 keras
pip install keras
- 安装 opencv
pip install opencv-python
安装完所需依赖后启动 welcome.py脚本
通过GPU启动完成