安装CUDA、CUDNN【GPU使用】

  • Post author:
  • Post category:其他




共需要安装以下几个部分:

1、nvidia驱动

2、CUDA

3、cuDNN

4、如果使用tensorflow,则安装对应的tensorflow-gpu版本



一、安装驱动

驱动这就不说了,直接安装最新的版本



二、安装CUDA

首先检查电脑支持的GPU版本

cmd命令行输入:nvidia-smi

在这里插入图片描述

这里cuda最高支持11.6,因此安装的时候要小于这个版本

CUDA的版本决定了CUDNN以及后面要安装的tensorflow-gpu的版本



第一步:下载CUDA


下载链接


在这里插入图片描述


安装步骤:


在这里插入图片描述

在这里插入图片描述


检测是否安装好:


运行CMD程序,输入nvcc -V,显示如下,表示安装成功:

在这里插入图片描述



三、下载安装CUDNN


下载链接

在这里插入图片描述

提示:可以复制链接到迅雷下载,无需注册账号

在这里插入图片描述

在这里插入图片描述


检验CUDNN是否安装成功:


在这里插入图片描述



版权声明:本文为m0_46423830原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。