windows系统安装深度学习框架tensorflow-gpu

深度学习环境搭建:Python35(64)+tensorflow-gpu

一 .深度学习框架对比

                          

                                                                                       表1-1  主流深度学习框架

                        



二 .深度学习环境搭建

         在windows环境下搭建tensorflow-gpu开发环境

1 .前期准备

    (1)显卡:Nvidia系列显卡(目前只有Nvidia旗下显卡支持GPU加速),

查看哪些显卡支持GPU加速,链接:https://developer.nvidia.com/cuda-gpus

      (2)  python3.5+(64bit)和一个python的IDE。

   (3)cuda安装包和cudnn加速库。

2.安装python3.5(Anaconda)版本 

       python下载地址:https://www.python.org/downloads/windows/直接下载exe文件到本地安装,安装之后需要添加环境变量path:你的python的安装目录\;你的python的安装目录\Scripts\。然后安装一个python的IDE,推荐使用Pycharm,链接:http://www.jetbrains.com/pycharm/download/#section=windows,安装Pycharm时直接勾选自动添加环境变量,之后即可,Pycharm会自动找到系统已存在的python版本。

3. 安装tensorflow-gpu版本      

    Tensorflow有cpu和Gpu两个版本,后者支持并行加速。

     (1)安装tensorflow-gpu。在命令行下定位到 .python/Scripts,然后输入 pip install tensorflow-gpu,之后系统自动获取需要的组件,安装完毕,具体步骤见下图。

                  

到这里已经安装好了tensorflow-gpu版本,但是还不完整,不能进行并行加速。

4. 需要安装一个cuda包和一个cudnn库

       CUDA(Compute unified Architecture),是英伟达(Nvidia)公司推出的一种基于新的并行编程模型和指令架构的通用并行计算结构,它能利用英伟达GPU的并行计算引擎,比CPU更高效的解决许多复杂计算任务。Cuda8.0下载链接:https://developer.nvidia.com/cuda-downloads

           

下载到本地之后,由于是exe文件,直接安装即可。

CudnnCudaDeep Neural Network library它是用于深度神经网络基本计算的 GPU 加速库。Cudnn链接:https://developer.nvidia.com/cudnn。下载这个安装包需要注册并且填一些问卷方可下载,下载的相关包不用安装,解压后拷到cuda路径对应的文件夹下面,再配置path环境变量即可。

(1)  解压后的cudnn下面有3个文件夹bin,libinclude,只需要把这三个文件夹下的文件复制到cuda相对应的文件夹下即可。

(2)  以我的为例,安装完cuda之后CUDA_PATHC:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v8.0,但是这样不能直接访问到该路径下的binlib下的程序包,在Path中加上这两个路径即可。即C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v8.0\bin

C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v8.0\lib


5  tensorflow测试小程序

    











  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值