二、在Windows下安装TensorFlow GPU版本教程(超级详细哦)

  • 1、确认显卡是否支持CUDA
  • 1.1、查看支持CUDA的显卡
  • 1.2、查看系统信息
  • 1.3、查看显卡信息
  • 2、安装CUDA
  • 2.1、下载并安装CUDA
  • 2.2、设置解压缩安装程序的暂存目录
  • 2.3、同意协议并继续
  • 2.4、选择精简安装选项
  • 2.5、警告未安装Visual Studio
  • 2.6、下载CUDA界面
  • 2.7、确认是否安装此设备软件
  • 2.8、确认安装
  • 2.9、安装完成
  • 3、安装cuDNN
  • 3.1、下载cuDNN
  • 3.2、加入会员
  • 3.3、进入下载页面
  • 3.4、查看下载后的文件
  • 3.5、复制到其他目录
  • 3.6、创建tools目录,并粘贴之前复制的文件
  • 3.7、已复制/粘贴到tools目录,如下图所示:
  • 3.8、查看cudnn64_5.dll
  • 4、将cudnn64_7.dll存放的位置加入Path环境变量
  • 4.1、编辑系统环境变量
  • 4.2、单击“环境变量”按钮
  • 4.3、编辑环境变量
  • 4.4、编辑用户变量
  • 5、在Anaconda建立TensorFlow GPU虚拟环境
  • 5.1、重新启动“命令提示符”程序,并切换到工作目录
  • 5.2、在Anaconda建立TensorFlow GPU虚拟环境
  • 5.3、启用TensorFlow GPU虚拟环境
  • 6、安装TensorFlow GPU版本
  • 7、安装Keras


1、确认显卡是否支持CUDA

1.1、查看支持CUDA的显卡

网址:http://developer.nvidia.com/cuda-gpus

GPUtil需要安装什么库_虚拟环境

1.2、查看系统信息

GPUtil需要安装什么库_tensorflow gpu安装教程_02

1.3、查看显卡信息

GPUtil需要安装什么库_虚拟环境_03


由上图知,本电脑的显卡为GeForce 940MX,支持CUDA软件,如下图所示:

GPUtil需要安装什么库_虚拟环境_04

2、安装CUDA

2.1、下载并安装CUDA

网址:https://developer.nvidia.com/cuda-downloads 安装步骤如下图所示:

GPUtil需要安装什么库_tensorflow gpu安装教程_05


下载后双击安装,如下图所示:

GPUtil需要安装什么库_GPUtil需要安装什么库_06

2.2、设置解压缩安装程序的暂存目录

GPUtil需要安装什么库_GPUtil需要安装什么库_07

2.3、同意协议并继续

GPUtil需要安装什么库_虚拟环境_08

2.4、选择精简安装选项

GPUtil需要安装什么库_虚拟环境_09

2.5、警告未安装Visual Studio

GPUtil需要安装什么库_环境变量_10

2.6、下载CUDA界面

GPUtil需要安装什么库_tensorflow gpu安装教程_11

2.7、确认是否安装此设备软件

注意:在安装之前关闭杀毒软件,否则会出现错误,安装失败。

GPUtil需要安装什么库_环境变量_12


如果安装过程中出现了上述的安全信息框,我们直接点击“安装”按钮即可。

2.8、确认安装

GPUtil需要安装什么库_环境变量_13

2.9、安装完成

GPUtil需要安装什么库_环境变量_14

3、安装cuDNN

3.1、下载cuDNN

网址:https://developer.nvidia.com/cudnn

GPUtil需要安装什么库_CUDA_15

3.2、加入会员

GPUtil需要安装什么库_GPUtil需要安装什么库_16


进入注册步骤:

GPUtil需要安装什么库_CUDA_17

3.3、进入下载页面

GPUtil需要安装什么库_虚拟环境_18

3.4、查看下载后的文件

GPUtil需要安装什么库_环境变量_19

3.5、复制到其他目录

GPUtil需要安装什么库_环境变量_20

3.6、创建tools目录,并粘贴之前复制的文件

GPUtil需要安装什么库_GPUtil需要安装什么库_21

3.7、已复制/粘贴到tools目录,如下图所示:

GPUtil需要安装什么库_tensorflow gpu安装教程_22

3.8、查看cudnn64_5.dll

GPUtil需要安装什么库_环境变量_23

4、将cudnn64_7.dll存放的位置加入Path环境变量

4.1、编辑系统环境变量

GPUtil需要安装什么库_CUDA_24

4.2、单击“环境变量”按钮

GPUtil需要安装什么库_GPUtil需要安装什么库_25

4.3、编辑环境变量

GPUtil需要安装什么库_CUDA_26

4.4、编辑用户变量

GPUtil需要安装什么库_环境变量_27

5、在Anaconda建立TensorFlow GPU虚拟环境

5.1、重新启动“命令提示符”程序,并切换到工作目录

GPUtil需要安装什么库_环境变量_28

5.2、在Anaconda建立TensorFlow GPU虚拟环境

输入conda create --name tensorflow-gpu python=3.5 anaconda命令

如下图所示:

GPUtil需要安装什么库_tensorflow gpu安装教程_29


环境加载完成后会出现下面界面:

GPUtil需要安装什么库_虚拟环境_30


安装完成的界面如下:

GPUtil需要安装什么库_CUDA_31

5.3、启用TensorFlow GPU虚拟环境

在命令行中输入:activate tensorflow-gpu,执行结果如下:

GPUtil需要安装什么库_虚拟环境_32

6、安装TensorFlow GPU版本

在命令行中输入:pip install tensorflow-gpu,执行结果如下:

GPUtil需要安装什么库_环境变量_33

7、安装Keras

在命令行中输入:pip install keras,执行结果如下:

GPUtil需要安装什么库_tensorflow gpu安装教程_34


亲!你还欠我一评价,请给予个好评,一点点码出的文字,很不容易。