配置成功版本:win10 x64+GTX 950M+CUDA 10.0+cudnn v7.6.5+Anaconda 3.5+Tensorflow1.14

注意:需要安装VS 2015及以上版本

       虽然现在tensorflow 2.x 已经出了,相比1.x 功能更好,对于新手来说更容易入门,但网络上关于1.x的资源也很多,所以也还在使用,本文主要介绍 1.x 的安装步骤。小伙伴们注意哦,文末有tensorflow2.x的安装介绍。

        1. 硬件要求

       先检查自己的电脑版本硬件是否满足要求,CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。可以到官网查看对应显卡的计算力https://developer.nvidia.com/cuda-gpus,下图为一部分显示(我的是GTX 950M的,算力为5.0)

tensorflow用cpu和显卡对比 tensorflow对显卡要求_官网

检查自己的显卡版本号,可以在终端输入nvidia-smi查看,也可以在自己的NVIDIA控制面板中——系统信息——组件里查看,我的版本是NVIDIA 10.2,只要在你的版本之下(包括你的版本)的CUDA都可以,但是,需要看官网给出的配置,来安装对应版本的tensorflow.

tensorflow用cpu和显卡对比 tensorflow对显卡要求_CUDA_02

       2. CUDA下载及安装

      到官网https://developer.nvidia.com/cuda-toolkit-archive去下载对应的版本,注意,选择本地下载。当然官网已经更新出更高版本了,可以根据自己的需求来下载配置。

tensorflow用cpu和显卡对比 tensorflow对显卡要求_官网_03

       安装:直接运行后,选择自定义安装。确保你之前已经安装了VS。

       把Visual Studio Integration和NVIDIA Geforce Experience去掉,如果下载的CUDA版本低于你电脑的显卡驱动版本,就需要去掉,不然会配置出错,如图设置,然后就一路向下,直到安装完成。安装路径如果C盘空间足够,尽量选择默认路径安装。

tensorflow用cpu和显卡对比 tensorflow对显卡要求_CUDA_04

完成后,运行cmd命令窗口,输入nvcc -V,检测是否安装成功

                                 

tensorflow用cpu和显卡对比 tensorflow对显卡要求_tensorflow用cpu和显卡对比_05

下载cuDNN,到官网下载,需要注册账号,根据对应的CUDA版本来下载,https://developer.nvidia.com/rdp/cudnn-download

                                               

tensorflow用cpu和显卡对比 tensorflow对显卡要求_CUDA_06

将cuDNN解压出来的文件夹下的文件复制到所安装的CUDA对应的文件下:

tensorflow用cpu和显卡对比 tensorflow对显卡要求_tensorflow用cpu和显卡对比_07

接下来添加环境变量:

tensorflow用cpu和显卡对比 tensorflow对显卡要求_官网_08

 3. tensorflow安装

        先安装Anaconda,勾选添加环境变量,一直下一步,直到安装完成。然后打开cmd运行,创建conda环境:

        创建一个叫tensorflow-gpu的环境:

conda create -n tensorflow-gpu pip pythnotallow=3.6

 等待安装完成之后,激活环境 activate tensorflow-gpu, 然后安装tensorflow

                              

tensorflow用cpu和显卡对比 tensorflow对显卡要求_tensorflow用cpu和显卡对比_09

pip install tensorflow-gpu==1.14

 等待安装:

                     

tensorflow用cpu和显卡对比 tensorflow对显卡要求_tensorflow_10

完成之后测试是否安装成功:

                                       

tensorflow用cpu和显卡对比 tensorflow对显卡要求_官网_11

                                       

tensorflow用cpu和显卡对比 tensorflow对显卡要求_官网_12

                                      

tensorflow用cpu和显卡对比 tensorflow对显卡要求_深度学习_13

 

ps: 以上软件有需要的小伙伴私聊我哦ε≡٩(๑>₃<)۶ 一心向学


下载最新版的Anaconda(CPU安装)。下载最新版本的Miniconda(GPU安装),安装Visual C++,安装完成后需要重启计算机。(注:硬件要求同上)

驱动版本注意:NVIDIA驱动程序需410.x 或更高版本,以下两个是依赖库的版本要求,不需要手动安装。
CUDA的版本需要是10.0
cudnn版本号需要不小于7.4.1

第一步:安装Miniconda   第二步:配置conda使用国内源(不会配置的可以直接下载文件放在相应位置)

第三步:打开Miniconda的命令行(Prompt),输入命令,等待安装完成即可。

conda install tensorflow-gpu==2.0.0