windows下安装gpu版本的keras\tensorflow

windows下安装gpu版本的tensorflow\keras

我的环境配置：

• Anconda3（python3.6）
• Nvidia GTX 1070
• Visual Studio 2015 Update 3
• windows 10

---

显卡说明

• 如果您的显卡是非NVIDIA公司的产品或是NVIDIA GTX系列中型号的第一个数字低于6或NVIDIA的GT系列，都不建议您采用此类显卡进行加速计算，例如NVIDIA GT 910、NVIDIA GTX 460 等等。
• 如果您的显卡为笔记本上的GTX移动显卡（型号后面带有标识M），那么请您慎重使用显卡加速，因为移动版GPU容易发生过热烧毁现象。
• 如果您的显卡，显示的是诸如 HD5000,ATI 5650 等类型的显卡，那么您只能使用CPU加速。
• 如果您的显卡芯片为Pascal架构（NVIDIA GTX 1080,NVIDIA GTX 1070等），您只能在之后的配置中选择CUDA 8.0

安装步骤

1.下载并安装python环境

python环境建设推荐使用科学计算集成python发行版Anaconda，注意：windows版本下的tensorflow暂时不支持python2.7.

下载地址：Anaconda

2.下载并安装Microsoft Visual Studio

CUDA编译器为Microsoft Visual Studio，版本从2010-2015，cuda8.0仅支持2015版本，暂不支持VS2017，本文采用Visual Studio 2015 Update 3。 贴出迅雷热链：

ed2k://|file|cn_visual_studio_professional_2015_with_update_3_x86_x64_dvd_8923256.iso|7745202176|DD35D3D169D553224BE5FB44E074ED5E|/

3.下载并安装CUDA

官网上只有cuda_9.1.85版本，但是由于的我的显卡芯片为Pascal架构（NVIDIA GTX 1080,NVIDIA GTX 1070等），所以只能安装cuda8.0.
在下载之后，按照步骤安装，不建议新手修改安装目录，同上，环境不需要配置，安装程序会自动配置好。

4. 加速库CuDNN

从官网下载需要注册 Nvidia 开发者账号，网盘搜索一般也能找到。 Windows目前最新版v6.0，但是keras尚未支持此版本，请下载v5.1版本，即 cudnn-8.0-win-x64-v5.1.zip。 下载解压出来是名为cuda的文件夹，里面有bin、include、lib，将三个文件夹复制到安装CUDA的地方覆盖对应文件夹，默认文件夹在：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\8.0

5.安装tensorflow

在CMD命令行中输入：

pip install -v tensorflow-gpu==1.2.1

安装tensorflow-gpu_1.2.1版本，如果不指定版本的话，默认安装最新的1.4.0版本，然而它与cuda8.0版本不对应。由于我的python环境（3.6）与其他tensorflow-gpu版本不兼容，所以我安装了1.2.1版本。
安装完成后可以验证tensorflow是否安装成功,在命令行中输入Python命令进入Python变成命令行环境：

>>> import tensorflow
>>>

安装成功啦！

6.安装keras

在CMD命令行中输入：

pip install keras

安装完成后进入python环境，输入：

>>> import keras
Using Tensorflow backend.
>>>

即为安装成功！

注意

安装过程中可能出现以下问题：
1. install tensorflow出现

Exception:  Traceback (most recent call last):    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\basecommand.py", line 215, in main      status = self.run(options, args)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\commands\install.py", line 335, in run      wb.build(autobuilding=True)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\wheel.py", line 749, in build      self.requirement_set.prepare_files(self.finder)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\req\req_set.py", line 380, in prepare_files      ignore_dependencies=self.ignore_dependencies))    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\req\req_set.py", line 554, in _prepare_file      require_hashes    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\req\req_install.py", line 278, in populate_link      self.link = finder.find_requirement(self, upgrade)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 465, in find_requirement      all_candidates = self.find_all_candidates(req.name)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 423, in find_all_candidates      for page in self._get_pages(url_locations, project_name):    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 568, in _get_pages      page = self._get_page(location)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 683, in _get_page      return HTMLPage.get_page(link, session=self.session)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 811, in get_page      inst = cls(resp.content, resp.url, resp.headers)    File "C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py", line 731, in __init__      namespaceHTMLElements=False,  TypeError: parse() got an unexpected keyword argument 'transport_encoding'   

解决办法：输入命令`

conda install -c anaconda html5lib

2.如果上述方法没有解决问题，尝试按照报错内容直接修改了C:\ProgramData\Anaconda3\lib\site-packages\pip\index.py line 731 的内容，即注释掉“transport_encoding=encoding,”
再重新安装tensorflow.

3.import tensorflow出现
Exception：can not find cudnn64_6.dll
去cuda安装目录下 NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\下查看此文件是否存在，如果存在但是版本号不对应，应该是cuda、cudnn、tensorflow版本不对应造成的（如果安装cuda8.0则cudnn也必须是8.0版本）。重新卸载安装即可。

---