美文网首页
如何安装 TensorFlow (Windows 环境下)

如何安装 TensorFlow (Windows 环境下)

作者: 智慧与力量的贤者 | 来源:发表于2017-10-14 15:37 被阅读0次

    前置检查

    • 已安装 Python 2.5 及以上版本(建议安装 3 以上版本)
      请自行到官网下载安装。
    • 已安装 Numpy
      获取地址(请根据自己的系统及需求下载)
      注1:也可以在 cmd 中使用 pip 安装(前提是已经安装 pipPython3 及以上版本安装好以后就自带的),在命令行中输入:
      pip3 install numpy
      然后等待安装成功。
      注2:安装过程中若出现问题请到 这里 查询。
    • 已安装 VisualStudio 2015
      这个教程很多,随便百度一下就会出来很多。

    使用 pip 安装 TensorFlow

    • 安装 GPU 版 (运行速度更快,推荐,但只有 NVIDIA 的显卡才能安装)
      cmd 中输入 pip install tensorflow-gpu
    • 安装 CPU
      cmd 中输入 pip install tensorflow

    安装 cuda

    划重点!!!! 在安装之前,请务必关闭 360安全卫士/腾讯管家 等安全软件!!!

    • 安装之前请先到 这里 查询你的电脑显卡是否支持 cuda
    • 官网 下载自己需要的版本
      注:建议下载本地版,网络版会从网页获取数据,可能会失败;
    • 双击 .exe 文件开始安装
    • 如果选择自定义安装,建议使用默认路径。
    • 安装完成后,可在 cmd 中输入 nvcc -V 来查看当前已安装 cuda 的版本信息,如果输出了版本信息,说明安装成功。

    下载 cudnn

    • 同理,到官网下载对应你的 cuda 版本的 cudnn
      注1:如果安装的是 cuda8 ,请下载 cudnn v6,因为使用 pip 安装的是最新版的 TensorFlow,而 TensorFlow1.3 就已经不再支持 cudnn v5,而支持 cudnn v6
      注2:下载完成后是一个压缩包,解压后有三个文件,如图
      cudnn 解压后
      请将三个文件夹一并拷入安装 cuda 中的 cuda 文件夹中,如:
      C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8
      ,与命名相同的文件夹合并,如图:
      cuda 文件夹
    • 检测系统环境变量 PATH 是否配置好,例如:
      C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
      C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

    检查是否能正常使用

    • 在 菜单->程序->Python 中,找到 Python 自带的编辑器 IDLE,打开,并输入:
      import tensorflow as tf
      回车,若成功,则安装成功。

    补充:

    1. 简单矩阵乘法 测试示例代码:
    import tensorflow as tf
    a = tf.random_normal((100, 100))
    b = tf.random_normal((100, 500))
    c = tf.matmul(a, b)
    sess = tf.InteractiveSession()
    sess.run(c)
    

    输出:

    output

    相关文章

      网友评论

          本文标题:如何安装 TensorFlow (Windows 环境下)

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