TensorFlow2.1菜鸟安装日记

情况简介

本人电脑上预安装的TensorFlow2.1，打算再安装一个TensorFlow-gpu2.2(后面了解到TensorFlow2.1版本后默认同时安装GPU与CPU，可以通过自己定义使用CPU/GPU)。。。这篇博文就记录一下比较实用的注意事项，避免后面同学走弯路。

TensorFlow安装前准备工作

TensorFlow-GPU安装使用前，需要电脑是英伟达-NVIDIA显卡。
参照 TensorFlow官网关于显卡支持内容：https://tensorflow.google.cn/install/gpu?hl=zh_cn

1. 硬件要求
   支持以下带有 GPU 的设备：CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。请参阅支持 CUDA 的 GPU 卡列表。查看网址
2. 软件要求
   必须在系统中安装以下 NVIDIA® 软件：
   NVIDIA® GPU 驱动程序：CUDA 10.1 需要 418.x 或更高版本。
   CUDA® 工具包：TensorFlow 支持 CUDA 10.1（TensorFlow 2.1.0 及更高版本）下载网址
   CUDA 工具包附带的 CUPTI。
   cuDNN SDK（7.6 及更高版本）下载网址
   （可选，本人未安装）TensorRT 6.0，可缩短用某些模型进行推断的延迟时间并提高吞吐量。

安装前特别注意事项

CUDA+cudnn+TensorFlow：三者需要版本对应。
本人安装CUDA10.1+cudnn7.65+TensorFlow2.1 完美运行

截止目前为止，CUDA10.2尚未支持，请勿安装，避免后续不必要的麻烦。

CUDA检查与安装

Anaconda Prompt命令行输入nvcc -V 可查看CUDA版本，注意中间有一个空格。

CUDA® 工具包：TensorFlow 支持 CUDA 10.1（TensorFlow 2.1.0 及更高版本）下载网址
在上面网站可以下载需要安装包，无脑安装即可。最后默认的安装路径会在C:\Program Files\NVIDIA GPU Computing Toolkit

CUDNN安装

cudnn安装并非安装包，下载后是一个压缩文件。解压后（可以改名方便后续添加环境变量），放在CUDA的安装路径下即可。参考题主的默认路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
如下图所示：

配置环境变量

我的电脑——属性——高级系统设置——环境变量——系统变量——Path——编辑——新增：添加Cudnn.bin路径（另外两个题主已经自动配置了，若无请自行添加）
参考题主默认路径：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn765\bin
最终如图：应为三个CUDA路径

TensorFlow安装

完成了上述CUDA、cudnn安装后，即可安装TensorFlow.

//cmd命令行下，将默认安装最新版
pip install tensorflow 

//cmd命令行下，将安装旧版
pip install tensorflow==2.1

查看与检验是否成功安装

cmd命令行下分别键入：

ipython
import tensorflow as tf
tf.version

再键入 tf.test.is_gpu_available()

等待一长串信息输出后…最后一行为True,恭喜你完成了TensorFlow的安装

补充：CUDA版本不正确处理办法

CUDA版本不正确处理办法：WindowsPC，可以通过控制面板——程序与功能——卸载软件。
卸载包含错误版本的CUDA的5个文件，重新在上述链接下载正确的CUDA版本以及匹配的cudnn版本。

来吧，import tensorflow as tf 😃