Win10 Tensorflow GPU版 安装教程 超详细

一、安装Anaconda;

请先安装Anaconda，具体安装教程详见上篇文章
https://blog.csdn.net/weixin_39671962/article/details/110576433

二、查找要安装的版本；

首先查找自己电脑的GPU型号；然后在官网查找tensorflow型号以及它需要的环境

2.1查找电脑GPU型号

查找自己电脑的GPU型号步骤：此电脑 —> 管理 —> 设备管理器 —> 显示适配器 —> 自己的型号
在此，我的GPU型号是：NVIDIA GeForce RTX2060

2.2查看Tensorflow配置

在官网查看电脑GPU型号是否支持CUDA

查找要Tensorflow要安装的版本以及需要的配置https://www.tensorflow.org/install/source_windows
上图查过了，我的显卡能安装CUDA
每个版本的Tensorflow所需要的环境有所差异，请严格按照Tensorflow官网的配置环境进行安装，比如：
安装tensorflow_gpu-2.3.0版本：Python版本3.5-3.8，cuDNN版本7.4，CUDA版本10.1；
安装tensorflow_gpu-2.2.0版本：Python版本3.5-3.8，cuDNN版本7.4，CUDA版本10.1
…
选择你要的版本就好了，我选择以tensorflow_gpu-2.3.0版本为基准，所以我需要配套安装的配置为：Python版本3.5-3.8，cuDNN版本7.4，CUDA版本10.1；

搞清楚安装什么版本之后，我们接下来就开始安装吧：
tensorflow_gpu-2.3.0
Python3.8
cuDNN7.4
CUDA10.1

三、安装CUDA 10.1；

安装CUDA，上面我们说过，安装tensorflow_gpu-2.3.0需要安装的CUDA版本是10.1，
（1）官网下载CUDA 10.1 版本安装包，下载好安装包常规操作，一直点击下一步即可


（2）安装好了之后，查看系统配置环境变量是否有这两项
此电脑 —> 属性 —> 高级系统设置 —> 高级 —> 环境变量 —>查看是否有这两项CUDA_PATH和CUDA_PATH_V10_1

(3)测试是否安装成功
打开anaconda的terminal输入nvcc -V，出现下列则安装成功

四、安装cuDNN 7.4；

安装cuDNN，上面我们说过，安装tensorflow_gpu-2.3.0需要安装的cuDNN版本是7.4
（1）下载安装包
先要注册账号，因为我CUDA是10.1版本的，而CUDNN没有7.4，就下了7.5版的


（2）解压安装包
将cuDNN安装包解压出来的三个文件，复制到CUDA的同名文件中

（3）添加环境变量

五、安装Tensorflow GPU

打开anaconda的terminal输入：
安装python:

conda create -n tensorflow pip python=3.8

安装tensorflow_gpu-2.3.0

pip install --ignore-installed --upgrade tensorflow_gpu==2.3.0

激活tensorflow

activate tensorflow

六、Pycharm配置tensorflow环境

File->settings->Python interpreter->Add->Conda Environment->选Existing environment->OK

import tensorflow as tf

gpu_device_name = tf.test.gpu_device_name()
tf.test.is_gpu_available()
print(gpu_device_name)
from tensorflow.python.client import device_lib

# 列出所有的本地机器设备
local_device_protos = device_lib.list_local_devices()
# 打印
#     print(local_device_protos)

# 只打印GPU设备
[print(x) for x in local_device_protos if x.device_type == 'GPU']

安装完成，棒棒哒！