超详细Anaconda搭配CUDA及cuDNN安装教程（Win10平台）

1.前言

笔者近期开始研究AI相关资讯，于此踩了不少坑，因此秉持着学习如水杯一样，当满出来时就可以进行分享的心态，所以笔者目前于此阶段已算满出来可以分享，以下流程秘籍让各位可以少走弯路。

2.开发环境工具与工具安装

以下环境是在Windows 10下建置。

a.开发环境工具与介绍

i. Anaconda：

由于在AI学习中经常遇到不同的学习框架（TensorFlow、Caffe、Pytorch等等）需要不同的环境配置，如果在相同环境下进行安装配置，容易出现版本问题，最后环境会变得非常混乱。

因此笔者要先教大家如何于Window 10环境架设Anaconda进行使用，为什么使用Anaconda？因为Anaconda可以建立多个不同的虚拟环境，而使用者不管安装什么学习框架，虚拟环境内的配置都不会互相干扰，因此笔者利用Anaconda来建立不同版本的TensorFlow或者Python版本。

ii.CUDA：

CUDA是NVIDIA公司研发的一套运算平台。CUDA可利用图形处理单元（GPU）为AI学习框架进行提升运算效，目前主流的AI学习构架如：Tensorflow、Pytorch等等也都使用CUDA进行训练让学习速度加快。也是多亏了GPU的运算性能，才能让AI学习可以有今天的表现，看官们如果想更进一步了解，请至NVIDIA官方站点搜寻。

iii. cuDNN：

cuDNN是NVIDIA开发用来为深度神经网络进行GPU加速的函式库，cuDNN可支持通用的AI学习框架，如：Tensorflow、Pytorch、caffe等等。cuDNN函式库可让开发人员于深层神经网络中取得更好的性能，简单的安装既可让开发人员只需专注于设计和实现神经网络模型，并可利用GPU实现高效能算，看官们如果想更进一步了解，请至NVIDIA官方站点搜寻。

b. Anaconda下载安装

下载网址：https://www.anaconda.com/products/individual

下载最新的Anaconda Individual Edition（图1），并进行安装。

图1 

c.查询TensorFlow所需工具的对应版本

笔者后续是使用TensorFlow进行训练，因此可至https://tensorflow.google.cn/install/source#gpu找寻TensorFlow所需要的python、cuDNN、CUDA对应版本（图2），此示例使用tensorflow-2.6.0 GPU版本，因此对应的相关套件版本为：

Python:3.6 ~ 3.9

cuDNN:8.1

CUDA:11.2

图2 

d.下载与安装CUDA

i.示例程序使用NVIDIA GeForce RTX 3080的显卡，因此先查看显卡驱动版本，驱动版本为496.49（图3）

nvidia-smi

图3 

ii.使用以下网址查看目前显卡的驱动版本是否支持CUDA 11.2，图4可看到显卡的驱动能够支持CUDA 11.2

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cublas-resolved-issues

图4 

iii.登入NVIDIA网站https://developer.nvidia.com/cuda-toolkit-archive，选择CUDA 11.2版本（图5）。

图5 

iv.下载Windows 10 x86_64版本（图6）

图6 

v.安装CUDA软件

a）下载的软件进行执行安装，一开始会遇到解压缩项目，直接默认OK（图7）。

图7 

b）选择自定义安装（图8）

图8 

c）之前安装显卡时已安装过驱动因此Driver components及Other components”不勾选”（图9），后续一直按下一步，既可完成安装。

图9 

d）安装完毕（图10）

图10 

vi.设定环境变

a）数环境变数→系统变数→Path设定相关CUDA变数

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\cuda\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\libnvvp

b）数环境变数→系统变数→新增相关CUDA_PATH_V11_2变数

变数名称

CUDA_PATH_V11_2

变数值

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

c）数环境变数→系统变数→新增相关CUDA_PATH_V11_2变数

变数名称

NVCUDASAMPLES11_2_ROOT

变数值

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.2

e.下载与安装cuDNN

i.下载cuDNN

TensorFlow 2.6需要cuDNN 8.1版本，因此登入NVIDIA网站https://developer.nvidia.com/rdp/cudnn-archive，下载Download cuDNN v8.1.0（January 26th，2021），for CUDA 11.0,11.1 and 11.2（图11）并选择cuDNN Library for Windows（x86）（图12）。

图11 

图12 

ii.安装cuDNN

下载的档案进行解压缩，会得到cuda文件夹，文件夹内会有bin、include、lib文件夹，把里面的数据复制到先前CUDA安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2内各自的bin、include、lib文件夹中（图13），既可完成安装。

图13 

f.创建Anaconda虚拟环境并安装Python 3.9

命令提示字元中进行创建一个新的Anaconda虚拟环境：

i.创建名称为tensorflow且python版本为3.9的Anaconda虚拟环境

conda create -n tensorflow pip python=3.9

ii.创建时会安装相关packages，请输入”y”继续安装（图14）

图14 

iii.创建好后进行tensorflow虚拟环境（图15）

conda activate tensorflow

图15 

g.查询Anaconda下有多少个虚拟环境

i.在命令提示字元中输入以下指令进行查询

conda env list

ii.查询结果（图16）

图16

结语

透过以上的教学，相信各位已经能够成功透过Anaconda架设虚拟环境，并进行环境切换；后续笔者会教大家如何使用TensorFlow2.x进行SSD-MobileNet迁移训练与验证，有问题的看官们欢迎联系我，我们一同讨论。

喜欢我的帖子，请帮我按个”收藏”，我们下回见。

也可原文继续关注我哦~