【环境配置】CUDA和cuDNN安装

文章目录

  • 1.CUDA安装(有N卡,v10.0)
    • 什么是CUDA
    • 下载安装包
    • 安装步骤
  • 2.安装cuDNN(v7.5.0)
    • 什么是cuDNN
    • 下载安装包
    • 解压
    • 环境变量配置。

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

什么是CUDA

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

下载安装包

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

下载选项:
【环境配置】CUDA和cuDNN安装_第1张图片

安装步骤

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

2.安装cuDNN(v7.5.0)

什么是cuDNN

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

下载安装包

  • 进入官网后,免费注册一个账号后就可以开始下载。
    Download cuDNN v7.5.0 (Feb 21, 2019), for CUDA 10.0

解压

下载好后,解压得到名为cuda的文件夹,然后将解压缩后的文件复制到cuda安装路径下的cudnn文件夹(这个文件夹需要自己建)。如图所示:
在这里插入图片描述

环境变量配置。

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

你可能感兴趣的:(Deep,Learning)