Ubuntu 18.04上安装CUDA 11.4、cuDNN 8.2的详细指南

在开始之前，请确保您的系统满足安装CUDA 11.4和cuDNN 8.2的要求。您需要一个支持CUDA的NVIDIA显卡，并确保已安装最新的显卡驱动。以下是详细的安装步骤：
步骤1：安装显卡驱动
首先，您需要更新系统软件包并安装NVIDIA显卡驱动。打开终端并执行以下命令：

sudo apt update
sudo apt install nvidia-driver-470

步骤2：下载CUDA 11.4和cuDNN 8.2
接下来，您需要从NVIDIA官网下载CUDA 11.4和cuDNN 8.2的安装包。您可以按照以下步骤进行下载：

# 打开终端
# 下载CUDA 11.4安装包
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
# 下载cuDNN 8.2安装包
wget https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/v8.2.2/cudnn-11.4-linux-x64-v8.2.2.26.tgz

步骤3：安装CUDA 11.4和cuDNN 8.2
接下来，您需要安装下载的CUDA和cuDNN安装包。执行以下命令：

# 安装CUDA 11.4
sudo sh cuda_11.4.0_470.42.01_linux.run
# 解压cuDNN 8.2安装包
tar -xzvf cudnn-11.4-linux-x64-v8.2.2.26.tgz

步骤4：配置环境变量
在完成安装后，您需要配置环境变量以确保系统能够找到CUDA和cuDNN的相关文件。执行以下命令：

sudo nano /etc/environment

在打开的文本编辑器中，添加以下行：

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

然后保存并关闭文件。为了使更改生效，执行以下命令：

sudo source /etc/environment

步骤5：验证安装成功
最后，您可以通过运行以下命令来验证CUDA和cuDNN是否成功安装：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 ...#define CUDNN_MAJOR 8#define CUDNN_MINOR 2

如果输出结果显示CUDNN_MAJOR为8，CUDNN_MINOR为2，则表示您的系统已成功安装CUDA 11.4和cuDNN 8.2。现在您可以开始使用您的GPU进行深度学习和其他计算任务了。