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，有问题欢迎打扰

转载于:https://www.cnblogs.com/xueadas/p/9758017.html