Linux系统中安装CUDA和cuDNN教程并检查安装成功

原创已于 2023-06-03 15:02:10 修改 · 1.5w 阅读

69 ·

CC 4.0 BY-SA版权

文章标签：

#linux #python

于 2022-07-18 14:17:36 首次发布

服务器专栏收录该内容

12 篇文章

订阅专栏

本文详细介绍了如何在Linux系统中安装CUDA和cuDNN。首先，通过`nvidia-smi`命令检查显卡驱动支持的CUDA版本，然后从CUDA官网下载并安装相应版本。安装完成后，配置环境变量，并验证CUDA安装是否成功。接着，下载与CUDA匹配的cuDNN版本，解压并复制到CUDA安装路径中，最后确认cuDNN安装成功。

安装CUDA

驱动适配

首先要查看服务器中显卡驱动支持的最大cuda版本。
在服务器中输入：nvidia-smi，得到如下版本信息。
在这里插入图片描述
第一行可以看到，最大支持cuda 11.6。

安装包下载

在cuda官网选择合适的版本
在这里插入图片描述
根据如图选择配置。

安装成功会出现安装路径。

配置环境变量

打开配置文件

vi /etc/profile

在配置文件末尾加上（注意这里的/usr/local/cuda-11.5路径和上面安装截图里面的路径一直）

export PATH=//usr/local/cuda-11.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.5/lib64$LD_LIBRARY_PATH

source一下配置文件

source /etc/profile

检查是否安装完成
使用nvcc -V检查CUDA是否安装成功，出现一下提示代表安装成功。
在这里插入图片描述
编译并执行CUDA样例程序，出现pass代表CUDA和GPU运行正常。

cd /usr/local/cuda-11.5/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

在这里插入图片描述

安装cuDNN

到官网下载与CUDA匹配的
在这里插入图片描述
我是在本地下载好了后，上传到服务器首目录下，然后解压。

解压cuDNN文件，并进入解压出来的文件，拷贝文件到/usr/local/cuda-11.5中

	tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz
	cd cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive
	sudo cp lib/* /usr/local/cuda-11.5/lib64/
	sudo cp include/* /usr/local/cuda-11.5/include/
	sudo chmod a+r /usr/local/cuda-11.5/lib64/*
	sudo chmod a+r /usr/local/cuda-11.5/include/*

查看cuDNN版本，cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
在这里插入图片描述