【环境配置】CUDA和cuDNN安装

1.CUDA安装(有N卡,v10.0)

什么是CUDA

CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

下载安装包

注意:tensorflow2.0对应的cuda版本是10.0,绝对不能下载错。下载网页链接

下载选项:
【环境配置】CUDA和cuDNN安装

安装步骤

下载好后打开安装程序,安装步骤如下:
【环境配置】CUDA和cuDNN安装
设置如下
【环境配置】CUDA和cuDNN安装 点击下一步开始安装。
【环境配置】CUDA和cuDNN安装
之后就cuda就算安装成功了。

2.安装cuDNN(v7.5.0)

什么是cuDNN

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。CUDA看作是一个工作台,而cuDNN就相当于工作的工具。想要在CUDA上运行深度神经网络,就要安装cuDNN。

下载安装包

解压

下载好后,解压得到名为cuda的文件夹,然后将解压缩后的文件复制到cuda安装路径下的cudnn文件夹(这个文件夹需要自己建)。如图所示:
【环境配置】CUDA和cuDNN安装

环境变量配置。

在系统变量的Path里面添加下面的四个项目:缺一不可,且顺序要在最上面(为了避免与其他版本冲突):
【环境配置】CUDA和cuDNN安装