Win10下安装Tensorflow+Keras简记(GPU版本)

 作为门外汉,对深度学习和人工智能很感兴趣,期望能够将人工智能应用到实际工作中。经过前期的了解,决定使用Tensorflow和Keras进行学习,搭建开发环境的过程中参考了很多网络资料,也形成了一些自己的体会,于是有了本篇简纪。

        开发环境:Windows10 64-bit。(备注:会有很多坑,经常遇到莫名其妙的问题)

        开发语言:Python3.6,强烈推荐anaconda,可以避免很多麻烦。(备注:最新版的Tensorflow和Keras已经支持Python3.6版,很多文档说要3.5版的可以忽略)

        学习框架:Tensorflow,Keras。(备注:二者的相关介绍请自行搜索)

参考资料:1.windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速

                  2.Installing TensorFlow on Windows

                  3.Keras安装和配置指南(Windows)

        分CPU版本和GPU版本分别说明安装步骤和过程,本文继续说明GPU版本的安装。前提条件:请先检查自己的GPU是否支持CUDA,具体方案是上Nvidia官网查询。关于CPU版本,请参考前一篇文章Win10下安装Tensorflow+Keras简记(CPU版本)


为了方便,建议大家提前准备好用到的软件,主要有:VS2015,DXSDK_Jun10,cuda_9.0.176_win10.exe(Nvidia官网下载地址),cudnn-9.0-windows10-x64-v7.zip(Nvidia官网下载,需要先注册),Anaconda3-5.1.0-Windows-x86_64.exe (Anaconda官网)。


简要说明安装步骤如下:

1、安装Visual Studio 2015,CUDA需要用到。目前1.5版本的Tensorflow支持CUDA9.0,而CUDA9.0已经支持Visual Studio 2017,所以安装Visual Studio 2017也是可以的。建议不要更改安装路径,避免环境变量的问题。


2、安装安装DXSDK_Jun10.exe。如果报错,百度可以找到解决方案。


3、安装CUDA9.0。目前1.5版本的Tensorflow支持CUDA9.0。由于Tensorflow经常更新,建议大家选择CUDA版本以官方文档为准(参考资料2)。请注意,一定要按照官方说明选择版本,避免不必要的错误。我此前安装了CUDA8.0,然后又安装CUDA9.0,运行时出过错误。


4、验证CUDA安装是否成功。第一步:打开命令提示符,输入:nvcc -V,如果能看到版本信息,已经成功一半了,但是还没有完全成功;第二步:编译CUDA示例程序,在 c:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0 目录下,打开Samples_vs2015.sln文件(请选择对应自己对应的VS版本),打开后,在VS编译选项中将解决方案配置更改为Release和x64。然后点击菜单:生成→生成解决方案。只要前面DirectX SDK安装不出问题,一般情况下都会成功生成方案的。然后在c:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64\Release找到deviceQuery.exe文件,打开命令终端,定位到刚才的目录,运行 deviceQuery.exe,如果正常显示自己电脑的显卡信息,则CUDA安装成功。


5、安装cuDNN。解压前面提到的cudnn-9.0-windows10-x64-v7.zip,有bin、include、lib三个目录,将三个文件夹复制到安装CUDA的地方覆盖对应文件夹,默认文件夹在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0。

6、安装Anaconda3-5.1.0-Windows-x86_64.exe。安装完成后,参考前一篇文章Win10下安装Tensorflow+Keras简记(CPU版本)的步骤1~步骤3,创建tensorflow环境变量。


7、安装GPU版本的Tensorfow。打开Anaconda Prompt,输入如下命令:

C:>activate tensorflow

(tensorflow)C:>pip install --ignore-installed --upgrade tensorflow-gpu


8、验证Tensorflow是否安装成功。打开Anaconda Prompt,输入如下命令:

C:>activate tensorflow

(tensorflow)C:>python

        然后在输入以下代码:

>>> import tensorflow astf

>>> hello = tf.constant('Hello, TensorFlow!')

>>> sess = tf.Session()

>>> print(sess.run(hello))

如果不报错误,安装成功会输入以下内容:

Hello, TensorFlow!


9、安装Keras。启动Anaconda Prompt,输入以下命令:

C:>activate tensorflow

(tensorflow)C:>pip install keras

安装后验证keras是否安装成功,继续输入:

(tensorflow)C:>python

>>> import keras

没有报错,那么Keras就已经成功安装了。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,948评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,371评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,490评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,521评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,627评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,842评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,997评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,741评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,203评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,534评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,673评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,339评论 4 330
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,955评论 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,770评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,000评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,394评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,562评论 2 349

推荐阅读更多精彩内容