博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Windows 7上安装配置TensorFlow-GPU运算环境
阅读量:6677 次
发布时间:2019-06-25

本文共 2578 字,大约阅读时间需要 8 分钟。

Windows 7上安装配置TensorFlow-GPU运算环境

1. 概述

  在深度学习实践中,对于简单的模型和相对较小的数据集,我们可以使用CPU完成建模过程。例如在MNIST数据集上进行手写数字识别,我们可以使用CPU来完成,采用经典的LeNet-5模型只需要十几分钟就能完成(具体取决于电脑配置)。

  然而,为了达到更好的效果或者构建更复杂的模型,一般需要更大的数据集,例如经典的ImageNet数据集。假如使用CPU训练模型将显得无比吃力(将会花费很长长长长长时间),此时GPU就可以派上用场了,利用其强大的并行计算能力训练模型会变快很多。

  本次教程主要分为以下几个步骤:

        安装显卡驱动

        安装CUDA

        安装CuDNN

        安装TensorFlow-GPU

注意请确保已安装python(若未安装请参考其它教程),并且最好为python3.5或3.6版本。

由于NVIDIA网站有时下载速度慢,提供百度网盘下载链接:

  cuda_9.0.176_windows.exe 链接:https://pan.baidu.com/s/1Jye0s0t2kJ7VQaG9ej8SDw    提取码:lrha

  cudnn-9.0-windows7-x64-v7.5.0.56.zip: 链接:https://pan.baidu.com/s/1cBj4B-RKd9iMvow75OXG5A   提取码:vowx

2. 安装显卡驱动

首先检查自己的电脑显卡是否支持TensorFlow-GPU计算(即是否支持CUDA并行计算架构),参考如下网站:

https://developer.nvidia.com/cuda-gpus

  

如上截图显示为部分NVIDIA显卡的计算能力(Compute Capability),TensorFlow-GPU一般要求要达到3.0以上。

确认自己的电脑显卡支持TensorFlow-GPU后下载安装显卡驱动,下载网址:https://www.nvidia.cn/Download/index.aspx?lang=cn

下载与自己电脑显卡型号匹配的驱动并安装,安装过程直接“下一步”“下一步”“下一步”即可。

查看显卡型号:

  计算机——>右击“管理”——>设备管理器——>显示适配器——>NIVDIA XXXX 

3. 安装CUDA

  CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

  参照下图选择合适的CUDA和cuDNN,可以看出CUDA 9和cuDNN 7适合大部分(最近)的TensorFlow-GPU版本。

  

CUDA下载网址:https://developer.nvidia.com/cuda-toolkit-archive

 

如图所示,选择:CUDA Tookit 9.0——>Windows——>x86_64——>7(代表Windows 7)——>exe(local) ——>Download(1.4GB)

本次下载的程序为:cuda_9.0.176_windows.exe

注意:选择合适的系统win10,win8还是win7;推荐选择exe(local)版,network版在安装过程中下载东西,可能导致安装过程很长长长长......

安装:默认“下一步”即可完成安装

4. 安装cuDNN

下载cuDNN 7(为什么下载cuDNN 7,请看第3节),下载需要注册帐号,请自行注册,注册过程比较简单:

https://developer.nvidia.com/rdp/cudnn-download

 

选择cuDNN v7.5.0 for CUDA 9.0进行下载,本次下载后的文件为:cudnn-9.0-windows7-x64-v7.5.0.56.zip

解压该文件,并将bin、include和lib三个文件夹复制到CUDA安装目录下。

注意:提示文件夹重复时选择“合并”即可。

5. 安装TensorFlow-GPU

打开终端cmd或Anaconda Prompt,用如下命令进行安装:

pip install tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple

注意:此处指定了TensorFlow-GPU的版本为1.9

           -i https://pypi.tuna.tsinghua.edu.cn/simple 代表使用清华镜像,速度非常快;如果不加-i选项将从官网下载Tensorflow包,下载速度非常慢,喜欢慢的朋友请去掉-i选项

6. 测试安装成功与否

在终端进入python,可使用如下命令测试是否安装成功:

 

如上能成功输出TensorFlow版本,表明安装应该没问题。

TensorFlow会自行判断并调用空闲的GPU,调用时显示GPU信息。也可在代码中指定用GPU进行运算。

注意:若测试失败,请尝试安装1.8版本的TensorFlow-GPU(在第5节安装命令中将1.9替换成1.8)。

7. FAQ

a. 为什么nvidia控制面板打不开,单击“nvidia控制面板”没反应?

答:计算机“右击”—>管理—>服务和应用程序—>服务,找到NVIDIADisplayContainerLS项服务,“右击”,“启动”或“重新启动”

b. 提示关于“vc++”, c complier之类的错误,或经常出现“python已停止运行”“python已停止工作”等时,请安装vc++2015试试?

答:vc++2015下载地址:

https://www.microsoft.com/en-us/download/details.aspx?id=53587

下载后直接安装即可。

提供百度网盘下载地址:  链接:https://pan.baidu.com/s/1QNAJ-tmBvm5dn9yHetJ9cA   提取码:iqlt

 


 

 

如有其它问题,请在评论中留言,一起探讨解决!

如有其它问题,请在评论中留言,一起探讨解决!

如有其它问题,请在评论中留言,一起探讨解决!

 

 

 

转载于:https://www.cnblogs.com/bigcindy/p/10706496.html

你可能感兴趣的文章
Serlvet学习笔记之二—不同页面共享数据
查看>>
Css3 - 动画旋转
查看>>
[Angular 2] Handling Click Events with Subjects
查看>>
django-redis和redis-py
查看>>
开发中常用js记录(二)
查看>>
Codeforces 715A. Plus and Square Root[数学构造]
查看>>
nodejs REPL(交互式解释器)
查看>>
颠覆你的认知,带你领略史上最为齐全的微软黑科技之旅
查看>>
【转】EL表达式 (详解)
查看>>
Android文件上传
查看>>
无人超市对社会是福还是祸?
查看>>
12306、上海地铁、广发银行的选择 起底蚂蚁金服移动开发平
查看>>
说阿里是日本的公司 你咋不说腾讯是非洲的呢
查看>>
春运的高铁上 有华为和满满的信号一起陪你回家
查看>>
以顶级项目之姿进入Apache基金会,Apache Kibble有点厉害!
查看>>
昆仑万维半年营收近18亿 旗下移动游戏平台启用独立品牌
查看>>
候车室里的年货:一件件都是家的故事
查看>>
广东侨乡台山人“小年夜”喜用传统沙琪玛祭灶
查看>>
电子品汽车服饰成春节热门“租物” 女性偏爱 “租锦还乡”
查看>>
世界经济论坛报告:地缘经济局势日趋紧张将阻碍今年经济增长
查看>>