tensorflow-gpu Win10 环境搭建

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启动完成