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

WPG大大通

于 2023-06-12 00:00:00 发布

阅读量7.8k

点赞数 2

文章标签：深度学习 caffe tensorflow ai 学习

本文链接：https://blog.csdn.net/wpgddt/article/details/131104477

版权

本文介绍了如何在Windows10环境下，使用Anaconda创建虚拟环境，以及安装CUDA和cuDNN以支持TensorFlow等AI框架的GPU加速。详细步骤包括Anaconda的下载安装、CUDA和cuDNN的选择与安装，以及环境变量的设置，旨在帮助读者避免版本冲突和配置问题。

摘要由CSDN通过智能技术生成

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