zoukankan      html  css  js  c++  java
  • Windows环境Tensorflow-cpu/gpu安装教程

     

     

    一、 CUDA与cuDNN下载与安装(后续有安装caffe想法的同学建议执行,仅安装tensorflow可跳过此步骤)

    CUDA:

    CUDA是由Nvidia创建的并行计算平台和应用程序编程接口(API)模型。它允许软件开发人员和软件工程师使用支持CUDA的图形处理单元(GPU)进行通用处理 - 这种方法称为GPGPU(图形处理单元上的通用计算)。CUDA平台是一个软件层,可以直接访问GPU的虚拟指令集和并行计算元素,以执行计算内核。

    下载链接: https://developer.nvidia.com/cuda-toolkit-archive

     TIPS:为确保设备运行兼容性,建议选择CUDA Toolkit 8.0 GA1/GA2

                 安装过程中请将CUDA安装到原始默认目录

    cuDDN:

    深度神经网络库(cuDNN)是用于深度神经网络的GPU加速原语库。cuDNN为标准例程提供高度调整的实现,例如前向和后向卷积,池化,规范化和激活层。cuDNN是NVIDIA深度学习SDK的一部分。

    下载链接: https://developer.nvidia.com/cudnn

    TIPS:点击Download cuDNN

     

     Nvidia要求注册账户方可下载,没有账户请点击Login完成相关注册。

     

    同样为确保兼容性,建议点击Archived cuDNN Releases

     点击Download cuDNN v5 (May 27, 2016), for CUDA 8.0,下载对应8.0版本。

    二、Anaconda下载与安装

    下载链接https://www.anaconda.com/download/#windows

     

    安装过程直接下一步即可,不做赘述,当遇到下图界面时建议两个选项均打勾

     

    安装完成后开始菜单会生成Anaconda文件夹

    如图所示打开Anaconda Prompt,

    在命令行输入 anaconda search -t conda tensorflow

    在列出的安装源中寻找合适的版本下载,在此通过Platforms中找到支持win-64的版本下载

    1.dhirschfeld/tensorflow(仅安装CPU版本)

       通过anaconda show dhirschfeld/tensorflow命令,系统展示进一步安装代码

     

    上图可知通过输入

    conda install --channel https://conda.anaconda.org/dhirschfeld tensorflow命令即可完成自动安装下载

    2.jjh_cio_testing/tensorflow-gpu(安装GPU版本)(此方法可自动配置cuda,因此可直接跳过第一个大步骤)

    通过anaconda showanaconda/tensorflow-gpu命令,系统展示进一步安装代码

     上图可知,通过输入

    conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu

    命令即可完成自动安装下载

    三、验证

    下载完成后打开Jupter Notebook,新建文件输入import tensorflow as tf 验证是否安装成功

    如图所示没有报错则代表安装成功,第一次运行可能出现警告,但不影响使用

     

     通过

    sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

    验证GPU还是CPU模式,可查看运行完成中Jupter Notebook中设备调用过程判断tenorflow版本,如图所示运行的是GPU

     

    本人QQ:2636188875,有问题欢迎打扰

  • 相关阅读:
    asp.net mvc 下载文件 txt doc xsl 等等
    MySQL用户管理:添加用户、授权、删除用户、删除用户、修改用户密码
    Vue打包时Unknown plugin "external-helpers" in "xxx\.babelrc" at 0" 解决方案
    vue中Axios的封装
    Docker部署ASP.NET Core应用到Linux中bash脚本
    Linux CentOS7系统中安装Nodejs、cnpm、Git环境
    C#之获取PDF张数以及PDF转JPG
    C#Amr转MP3
    mysql 之C#使用insert批量插入时日期字段为null,插入报错解决方案
    Windows服务器共享文件,Linux服务器实现挂载共享文件
  • 原文地址:https://www.cnblogs.com/xueadas/p/9758017.html
Copyright © 2011-2022 走看看