一、CUDA与cuDNN下载安装

进行神经网络训练时，如果搭配显卡即使用GPU进行训练，速度会高于使用CPU进行训练。TensorFlow同时支持CPU和GPU训练，在下载安装时选择安装tensorflow而不是tensorflow-cpu，就可以使用GPU进行快速训练。而在安装TensorFlow之前需要下载安装CUDA与cuDNN。
CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台，它是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。
cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销，可以集成到更高级别的机器学习框架中。
CUDA和cuDNN版本需要与要安装的TensorFlow版本相对应，对应列表可以查看官网说明https://tensorflow.google.cn/install/source_windows，如下图所示：

截至2020年11月TensorFlow最新版本为2.4版本，但官网说明中未给出对应cuda版本，本文以安装2.3版本为例，下载安装cuda10.1与cudnn7.4。
cuda官网下载地址为https://developer.nvidia.com/cuda-toolkit-archive，选择CUDA Toolkit 10.1 update2，进入下载页面，选择操作系统类型及版本和安装类型（在线或离线），然后下载安装包，安装时选择默认选项即可。

cuDNN下载地址为https://developer.nvidia.com/cudnn-download-survey，进入该界面后需要先按照提示使用邮箱注册账号，网速会有些慢，注册完毕后还需要再填写调查问卷，填写提交之后进入下载页面https://developer.nvidia.com/rdp/cudnn-download，在选择版本时发现7.4版本中没有For CUDA10.1，与TensorFlow官网说明不符，本文选择cuDNN7最新版cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1，然后选择系统版本，下载压缩包。

下载完成后解压缩，将下图三个文件夹内的文件复制到CUDA安装目录下对应目录内即可。

二、Conda下安装TensorFlow

使用conda下载依赖包时默认的源地址速度较慢，所以需要更换成国内的源，一般是清华、中科大、阿里云等源，首先打开Anaconda Powershell Prompt进入命令行界面，然后输入如下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls yes

然后创建一个用于使用TensorFlow的环境，命令如下：

conda create --name tensorflow python=3.8.6

回车之后会下载包然后安装，成功安装后界面如下：

输入如下命令进入创建的环境中，下载TensorFlow：

conda activate tensorflow
pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

下载安装完成后，输入python进行解释器中，输入：

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())

若输出tensorflow版本号"2.3.0"和“GPU ture"，说明成功。

三、PyCharm使用搭建的tensorflow环境

PyCharm的简介和安装见本博客其他文章，打开PyCharm软件，选择New Project，进入如下界面：

选择Existing Interpreter，点击…图标，进入Add Python Interpreter界面，选择Conda Environment，点击…图标，选择刚才创建的tensorflow环境内的python.exe，最后点击OK。

返回创建项目界面，写入项目名称，点击create即可。
在main.py中输入如下测试代码，点击运行：

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())

输出如下图即创建项目成功。

参考文档

win10下CUDA和CUDNN的安装（超详细）！亲测有效！
nvidia驱动，cuda与cudnn的关系
win10+tensorflow2.3GPU+cuda11+cudnn8+python3.8编译(AVX2)
Anaconda换源
conda创建新环境
Anaconda查找包,下载包,更新包,删除包
经验：WIN10+GTX1070下，tensorflow2.3-GPU+CUDA10.1+cuDNN765的安装
在PyCharm中使用Anaconda环境

Windows10下使用Conda安装TensorFlow-GPU相关推荐

安装 | cmd（命令提示符）窗口下使用conda安装TensorFlow
博主github:https://github.com/MichaelBeechan 博主CSDN:https://blog.csdn.net/u011344545 之前写过两篇关于安装TensorF ...
Windows10下通过anaconda安装tensorflow
博主经历了很多的坎坷磨难才找到一个比较好的在win10下安装TensorFlow的方法: 首先需要说明的是如果你想通过Anaconda来安装tensorflow的话,首先要确认你的python的版本是 ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
【Python】Conda安装TensorFlow和PyTorch的GPU支持包
Conda基本操作这里的Conda可以包括Miniconda和Anaconda,是一个统称. 配置conda镜像: vim ~/.condarc 内容: channels:- defaults sh ...
win7下cuda8.0安装跑gpu版tensorflow
要用深度学习做目标检测,先尝试了caffe,这会又要熟悉tensorflow了,简单写下配置过程吧,挺简单的: 0.win7 X64系统 1.安装vs2013 2.安装Anaconda3(需要里面的p ...
Windows 7下通过anaconda安装tensorflow
本贴解决目的:Windows 7下通过anaconda安装tensorflow 今天说一下通过Anaconda安装tensorflow,并安装运行spyder编译器.网上很多教程不全,官网的也没有详细 ...
通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)
一. 安装环境 Ubuntu16.04.3 LST GPU: GeForce GTX1070 Python: 3.5 CUDA Toolkit 8.0 GA1 (Sept 2016) cuDNN v6 ...
python3.6.5安装tensorflow_Win10下用Anaconda安装TensorFlow(图文教程)
笔者之前在学习TensorFlow,也在自己的笔记本上完成了安装,在PyCharm中进行学习.但是最近为了使用python的科学计算环境,我把之前的环境卸载了,并用Anaconda重新安装了Tenso ...
conda安装tensorflow和conda常用命令
1. 在conda的一个环境下安装tensorflow 1)先查看它的各个版本: conda search tensorflow 或者 conda search tensorflow-gpu 2)选择 ...

Windows10下使用Conda安装TensorFlow-GPU