美文网首页
windows10+tensorflow-gpu+keras深度

windows10+tensorflow-gpu+keras深度

作者: ZzzZBbbB | 来源:发表于2019-03-09 16:51 被阅读0次

    工欲善其事,必先利其器,以下是自己花了不少时间从网上找经验和方法搭建的windows版本下的深度学习环境,作为一个小结;


    了解一下术语:

    CUDA(Compute Unified Device Architecture,统一计算架构):NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。

    cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。

    tensorflow:谷歌深度学习框架,理解成一个库就行了。

    keras:就是站在巨人肩膀(tensorflow)上的人,是一个兼容 tensorflow的神经网络高级包,类似于封装了tensorflow,给开发者使用的时候更加方便。

    Anaconda:可以便捷获取包且对包能够进行管理,同时对环境可以统一管理的发行版本。Anaconda包含了conda、Python在内的超过180个科学包及其依赖项。理解成管理python各种东西的就行。


    本机配置:
    系统:windows10.
    显卡:NVIDIA GeForce GTX 1060 6GB

    安装内容与顺序:
    1.visual studio 2017
    2.CUDA9.0
    3.cuDNN 7.0.5
    4.Anaconda
    5.tensorflow-gpu

    安装内容.png

    具体步骤:
    1.安装vs;(vs必须安装,CUDA是与其结合使用的)
    进入官网进行下载:
    https://visualstudio.microsoft.com/downloads/
    安装过程中勾选 C++组件

    2.安装CUDA
    查看显卡类型,查看显卡支持的CUDA的版本 NVIDIA控制面板=>帮助=>系统信息=>组件=>NVCUDA.DLL=>产品名称 (自己的是9.1,但是!!!!!要安装9.0!9.0!9.0!)
    CUDA下载地址:
    https://developer.nvidia.com/cuda-toolkit-archive

    3.配置CUDA
    电脑右键,打开属性==>高级系统设置==>高级==>环境变量,可以看到系统中多了两个环境变量:CUDA_PATH 和 CUDA_PATH_V9_0,
    添加如下几个环境变量:

    CUDA_BIN_PATH = %CUDA_PATH%\bin
    CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
    CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0
    CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
    CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
    
    在用户变量Path中添加:C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0
    
    命令行输入 set cuda 查看配置情况
    或者进入  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite目录,使用命令行运行bandwidthTest和deviceQuery,看到显卡信息和‘Result = PASS 表示配置成功
    
    

    3.下载cudnn
    https://developer.nvidia.com/rdp/cudnn-archive(需要注册账号才能下载)
    根据CUDA版本选择相应版本
    下载成功后,将里面的bin、lib、include文件夹中内容复制到CUDA安装目录的对应文件目录下
    CUDA默认安装目录为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

    4.安装Anaconda
    https://www.anaconda.com/distribution/
    一直按照默认下一步即可,但是其中有一个,勾上,表示添加到环境变量中

    添加到环境变量.png

    4.下载tensorflow-gpu
    可以创建一个环境专门用于深度学习:

    创建新环境:conda create -n tensorflow-gpu python=3.6.5
    激活/切换到该环节  activate tensorflow
    

    选择正确的tensorflow-gpu版本


    版本对应.png

    在该环境下的命令行输入
    pip install tensorflow-gpu==1.8.0
    (这里存在一个pip速度慢的问题,解决方案如下:
    到user目录中建立一个pip目录,然后新建pip.ini,例如:C:\Users\SONDER\pip\pip.ini
    ini中的内容为:
    [global]
    index-url = https://pypi.tuna.tsinghua.edu.cn/simple)

    5.下载keras
    pip install keras

    6.检查是否安装成功
    在命令行输入
    import tensorflow
    import kears
    没有报错 则成功


    安装过程的各种坑和注意事项:

    CUDA cudnn tensorflow keras的版本号务必要对应,否则会发生各种各种错误!!!
    这里自己之前踩的坑是装的9.1版本的cuda,但是最后安装完成后会出现各种问题,然后百度发现tensorflow-gpu不兼容9.1 9.2 所以只能重新卸载了在装,不过自己比较强迫症,重装了windows系统后全部装了一遍。

    ~~~~~
    那么,我就祝你万事胜意吧。
    ~~~~~

    相关文章

      网友评论

          本文标题:windows10+tensorflow-gpu+keras深度

          本文链接:https://www.haomeiwen.com/subject/passpqtx.html