1. TensorFlow-gpu安装时要与CUDA、CUDNN版本对应
参考网址:https://tensorflow.google.cn/install/source#linux
部分截图:
在这里插入图片描述

2.CUDA与CUDNN选择需要的版本进行下载
CUDA快捷下载网址:https://developer.nvidia.com/cuda-toolkit-archive
CUDNN无需注册下载网址:https://dl.ypw.io/ubuntu-environment/#cudnn

3. 有时会出现直接下载CUDA进行安装,结果出现NVIDIA安装失败的问题
解决方法:
1、更新Windows系统。
2、更新显卡驱动(可以下载驱动精灵进行驱动更新,方便快捷,但注意驱动精灵有捆绑安装,安装时注意勾掉安装其他软件的选项。此方法亲测有效。)

4. tensorflow-gpu安装较详细的教程,且比较安全的教程网址如下。
TensorFlow-GPU安装教程: https://zhuanlan.zhihu.com/p/37086409

前三点可作为第四点教程的参考补充。

更多推荐