• 网上有很多教程,但是找一个适合自己电脑的却经历了很多困难。现在详细地记录下来,方便后来人参考吧!
  • 本教程安装的使tensorflow的【GPU】版本,想要安装CPU版本的参考另一篇博客【CPU版Tensorflow

1、本人电脑配置

  • Windows10,64位
  • 显卡:GTX1050Ti

2、准备资源(安装包)

  • Anaconda(需要64位的!!)
  • VS2015社区版
  • CUDA9.0
  • cuDNN7.4

3、安装过程

3.1、安装VS2015

  • 安装cuda需要使用VC++的编译,所以需要安装VS,安装时只需要选择安装Visual C++即可。
  • 安装过程很简单,选择好安装组件和安装路径之后,一路【next】
  • VS2015 Update3下载地址

3.2、安装Anaconda

  • 安装Anaconda的步骤也很简单,唯一需要注意的是要安装【64位】版本
  • 安装过程很简单,选择好安装路径,一路【next】,需要注意的是中间又一部添加环境变量要勾上!!
  • Anaconda3-2018.12-Windows-x86_64下载地址

3.3、安装CUDA

0、【cuda_9.0.176_win10.exe下载地址

1、双击下载好的[cuda_9.0.176_win10.exe]文件开始安装

2、首先会进行系统检查,检验版本是否符合。

  • 根据网上教程,首先安装的CUDA8.0,后来系统检测出错,如下图:大意是我得显卡太好,CUDA8配不上
    64位Win10 安装 tensorflow、keras(GPU版)
    64位Win10 安装 tensorflow、keras(GPU版)
  • 然后下载CUDA9,并安装:
    • 系统检查无误
    • 同意软件许可协议
      64位Win10 安装 tensorflow、keras(GPU版)
    • 选择自定义安装
      64位Win10 安装 tensorflow、keras(GPU版)
    • 选择安装位置(记住,等会要用)
      64位Win10 安装 tensorflow、keras(GPU版)
    • 安装过程
      64位Win10 安装 tensorflow、keras(GPU版)
    • 安装完成
      64位Win10 安装 tensorflow、keras(GPU版)
    • 测试安装,管理员身份打开命令提示行,输入:nvcc -V,注意是大V!
      64位Win10 安装 tensorflow、keras(GPU版)
    • 安装完毕

3.4、安装cuDNN7

  • 下载cuDNN需要注册账号
  • Download cuDNN v7.4.1 (Nov 8, 2018), for CUDA 9.0下载地址
  • 解压下载好的cuDNN文件夹,里面有三个文件夹【bin、include、lib】,将这三个文件夹下的最后一级文件分别放入CUDA9安装的文件夹***意:不是用这三个文件夹直接复制的CUDA的安装目录。
  • 具体而言,如下:
    • D:\cuda\bin\cudnn64_5.dll —> D:\Program Files\NVIDIA\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
    • D:\cuda\include\cudnn.h —> D:\Program Files\NVIDIA\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
    • D:\cuda\lib\x64\cudnn.lib —> D:\Program Files\NVIDIA\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
      64位Win10 安装 tensorflow、keras(GPU版)

3.5、使用Anaconda创建虚拟环境

  • 1、使用管理员身份打开命令提示符
  • 2、创建名为tensorflow_GPU的conda环境:【conda create -n tensorflow_GPU pip python=3.5】
    64位Win10 安装 tensorflow、keras(GPU版)
  • 3、**tensorflow_GPU的conda环境:【activate tensorflow_GPU】
    64位Win10 安装 tensorflow、keras(GPU版)
  • 4、升级pip至最新版:【python -m pip install --upgrade pip】
    64位Win10 安装 tensorflow、keras(GPU版)
  • 5、安装GPU版tensorflow:【pip install --ignore-installed --upgrade tensorflow-gpu】
    64位Win10 安装 tensorflow、keras(GPU版)
  • 6、安装完毕,进行测试
  • 7、补充:如果你想使用Keras进行深度学习,可以在第5步之后继续安装keras包:【pip install keras -U --pre】
    64位Win10 安装 tensorflow、keras(GPU版)

4、测试

  • 1、为了清晰,可以重新打开一个命令行
  • 2、**tensorflow_GPU的conda环境:【activate tensorflow_GPU】
  • 3、进入python编译环境,输入:【python】
  • 4、输入如下代码(因为在Jupyter下编写的,所以注释掉了):
# import tensorflow as tf
# hello = tf.constant("hello,tf")
# with tf.Session() as sess:
#     print(sess.run(hello))
  • 经过一段时间(怎么感觉比非GPU版的还慢呢?),终于输出结果了:
    64位Win10 安装 tensorflow、keras(GPU版)

5、终于好了!!对你如果有帮助,点个赞,鼓励一下吧

相关文章: