TensorFlow GPU版本安装步骤-Windows系统

本文由在Windows下安装TensorFlow GPU版本教程-CSDN等整理得到。

1、确认显卡是否支持CUDA

1.1 查看显卡信息

打开“设备管理器”，查看显卡信息，如下图所示：

由上图知，本电脑的显卡为NVIDIA Quadro P400.

1.2 确认是否支持CUDA

网址：https://developer.nvidia.com/cuda-gpus
经查询可知，该显卡支持CUDA，如下图所示：

2、安装CUDA

2.1 下载CUDA

网址：https://developer.nvidia.com/CUDA-toolkit-archive
选择想要安装的版本，这里选择CUDA Toolkit 11.1.0 版本，如下图所示：

点击进去后选择Windows10系统，x86_64架构，以及网络安装方式，然后点击下载，如下图所示：

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

下载后双击安装，在弹出的对话框中设置压缩安装程序的暂存目录：

2.3 同意协议并继续

2.4 选择自定义安装

2.5 自定义安装选项

展开CUDA项，取消勾选“Visual Studio Integration”，避免在未安装Visual Studio的情况下出现安装错误，如下图所示：

收起CUDA项，展开“Driver components”项，确认“Display Driver”子项的新版本号与当前版本号，如若当前版本低于新版本，则按照默认设置，即勾选“Display Driver”，如若当前版本高于新版本，则取消勾选“Display Driver”，如下图所示：

然后点击下一步，后续按默认设置安装完成即可。

3、安装cuDNN

3.1 下载cuDNN

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