1.查看是否支持GPU
2.Anaconda下载与安装
3.Anaconda环境配置准备工作
4.在Anaconda中创建新环境
5.安装CUDA10.0
6.安装cudnn7
7.配置环境变量
8.安装tensorflow的GPU版本
9.测试是否安装成功
【补充说明】
【前言】1.引言(可不看,直接看2)
目前电脑都装有GPU,而我们往往用不着或者不能充分使用,所以我想着把它充分利用起来,在训练模型时还可以用CPU运行其他应用,这时候就需要装GPU版本的Tensoflow。Tensorflow的CPU版本安装相对容易,而tensorflow-gpu版本的安装相对来说较为复杂,需要考虑版本匹配,很容易出问题导致一遍遍重装。网上(b站、知乎、CSDN等)有很多教程,但他们往往无法说明安装Tensorflow的底层逻辑,而且python、Tensorflow、CUDA、CUDnn的版本更新很快,我们的电脑性能和硬件又不尽相同,这些都导致各种教程很容易过时和误导。这篇文章仍然很刻板,只教给大家适合很小范围内电脑配置的Tensorflow_gpu的安装方法,步骤也相当繁琐,但我会说明整个安装逻辑和注意事项,让大家根据这篇文章操作便可安装成功。
文章选用和总结了目前网上的各种经验,这里向各位前辈致敬!
2.安装环境 要求64位win10操作系统,GPU为GTX 960M或更高版本( 算力>3即可 ),下载好CUDA10.0、cuDNN7.6和Anaconda( 文末有网盘链接 )。 IDE(集成开发环境):Anaconda 、Pycharm 框架和解释器:Tensorflow-1.4.0-cp36-cp36m-win_amd64+python3.6 注: 上述两段可不必看懂 3.安装Tensorflow_gpu的底层逻辑 首先,我们安装Anaconda,它(最近的版本)可以为我们在windows系统上开设n个虚拟环境,我们开发、训练以及安装相关软件、配置相关环境(例如python)等都需要在某个环境中, TensorFlow_GPU版本也是要安装在这个环境中。环境可以设置多个,互不干扰,类似于我们电脑可以安装多个操作系统,可同时安装win10和win7一样。但tensorflow_gpu版本与cpu版本不同的是,它需要电脑gpu硬件以及软件的支持,那么 我们就要看自己的电脑硬件和软件是否支持安装gpu版本的tensorflow,或者说安装哪一个版本的tensorflow_gpu 。这样,在确保我们电脑支持的情况下,为支持我们安装的tensorflow_gpu版本可用,我们需要安装与电脑硬件软件匹配或容易获得安装包的CUDA和CUDnn。其中CUDnn是CUDA的类似于补丁的文件,所以他们两个版本需要匹配。 这样,电脑硬件和软件(指的是显卡驱动程序)、tensorflow、CUDA、CUDnn、Python它们五个就构成了一种微妙的版本适配关系,有一个装错你的tensorflow就测试不成功。下面进行傻瓜式操作。 4.安装顺序和文章安排 根据上边的介绍我们可以简单了解到,想要正确安装tensorflow_gpu,过程相当繁琐。整个教程将分三篇文章来说明整个过程,第一篇(即本篇)文章说明电脑软硬件的查看以及Anaconda的安装,第二篇说明CUDA、CUDnn以及Tensorflow的安装和测试,第三篇进行补充说明。已经搞明白怎么回事的读者也可以适当跳过不必要的步骤。 注:大图&多图预警,为了您更好的体验,建议电脑端查看 【安装过程】1.查看是否支持GPU
1.1 在搜索框搜索“控制面板”,找到NVIDIA控制面板(只要能找到NVIDIA控制面板,其他方式亦可),然后点左下角的“系统信息”,查看GPU型号,如下图: AMD的显卡是不支持的,检查显卡的计算能力请访问英伟达官网或者百度。 1.2 参考TensorFlow官网对安装GPU版本的需求说明,需要CUDA、cuDNN和英伟达GPU计算能力大于3的显卡(这里可以自行百度自己的显卡计算能力是否大于3,我的电脑GPU型号为GTX 960M,可以查到其计算能力为5,GTX 1050中数字前两位表示显卡是第几代,所以我的是第9代,个人认为如果你是10XX基本就不用查了,可以进行下一步了)。还可以查看GPU和软件的版本对应关系,好多人在这里踩了一个大坑,如下两图:
本文采用的的是tensorflow_gpu1.14.0、CUDnn7.6、CUDA10.0和Python3.6的组合,它要求显卡驱动版本大于410x(如上图)。但显然这个组合在上边两图是不成立的,但如果你的电脑硬件(型号)满足了,跟着教程走一定能安装成功。
1.3 升级显卡驱动程序
在步骤1.1的图中,红框右侧“驱动程序版本”需升级到410x以上,这个需要下载驱动精灵,升级显卡驱动。
至此,准备工作完成,可先跳至文末下载所有需要的安装包,再回到这里继续执行下面的步骤,如果显卡及驱动程序版本不能达到上述要求,建议去寻找其他教程,或者能够确定上面两图中的其他组合且能顺利获取相应的安装包的话可以继续执行下面步骤。
2.Anaconda下载与安装 下载地址: anaconda.com/products/individual docs.anaconda.com/anaconda/install/windows/中有针对各个系统安装过程的 英文步骤, 文末的链接里也有安装包,如已下载直接跳至 2.3 2.1 点击anaconda.com/products/individual网页中Download自动跳转 2.2 在跳转之后的页面中,根据自己的系统选择合适的版本(本文以Windows 32位系统为例进行说明,64位与之相同,记得下载上边那个64位的),点击进入下载2.3 双击安装包开始安装(注:除非有提示让以管理员身份安装,否则正常安装)
点击Next2.4 点击 I Agree
2.5 选择 Just Me, 再点击 Next 2.6 设置路径(不要有空格或“unicode”编码),点击Next 2.7 选择第一项(如果电脑中有其他版本的Anaconda或者Python,建议不选或者卸载后安装,因为这个软件所有的环境都会为你配置好,也有很多版本供你选择),目的是将Anaconda加入到环境中,避免以后手动添加。点击Install进行安装。(下图是64位的,和上边版本不一样,别产生疑惑,下边的就对了,上边的截图是我安装32位时的截图) 2.8 之后,程序进行安装 2.9 安装完成,点击Next 2.10 点击Next 2.11 最后,点击Finish即可 注 :两个方框可以不选至此,完成了Anaconda的安装
之后的步骤见第二篇文章《【人工智能&深度学习】Win10安装tensoflow_gpu详细教程(2)》
安装过程中经常出现的问题以及解决方法(补充说明)见第三篇文章《【深度学习】Anaconda 、Tensorflow安装和环境配置常见问题及解决方法》
安装包网盘链接:
链接:
https://pan.baidu.com/s/1OHCgiVq7SJRhqx54ZcS5NQ
提取码:sdxx
分享 点赞 再看三连留言