Ubuntu 22.04.1 LTS 安装 CUDA 11.0 和 cuDNN 总结

Ubuntu 22.04.1 LTS 安装 CUDA 11.0 和 cuDNN 总结

一、引言

随着深度学习和人工智能的快速发展，CUDA 和 cuDNN 在计算中发挥着越来越重要的作用。本文将指导你在 Ubuntu 22.04.1 LTS 上安装 CUDA 11.0 和 cuDNN，让你能够快速开始你的深度学习之旅。

二、安装前的准备

1. 检查显卡兼容性：首先，确保你的显卡支持 CUDA 11.0。你可以在 NVIDIA 官方网站上查找你的显卡型号，查看其是否支持 CUDA 11.0。
2. 安装显卡驱动：确保你已经安装了合适版本的 NVIDIA 显卡驱动。你可以在 NVIDIA 官网上下载并安装适合你显卡型号的驱动。

三、安装 CUDA 11.0

1. 下载 CUDA 11.0：访问 NVIDIA 官网，下载 CUDA 11.0 的安装包。根据你的系统和显卡选择合适的版本。
2. 运行安装包：在终端中运行 CUDA 安装包，并按照屏幕上的指示进行安装。注意选择正确的安装选项，如是否安装驱动等。
3. 配置环境变量：安装完成后，需要配置环境变量以便在终端中使用 CUDA。编辑 ~/.bashrc 文件，添加以下内容：

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

然后运行 source ~/.bashrc 使环境变量生效。

四、安装 cuDNN

1. 下载 cuDNN：访问 NVIDIA 官网，下载 cuDNN 的安装包。选择与你的 CUDA 版本兼容的 cuDNN 版本。
2. 解压安装包：在终端中解压下载的 cuDNN 安装包。
3. 复制库文件：将解压后的库文件复制到 CUDA 的库目录下。执行以下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

1. 配置环境变量：编辑 ~/.bashrc 文件，添加以下内容：

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

然后运行 source ~/.bashrc 使环境变量生效。

五、验证安装

1. 验证 CUDA 安装：在终端中运行 nvcc --version，如果看到 CUDA 的版本信息，表示 CUDA 安装成功。
2. 验证 cuDNN 安装：编写一个简单的深度学习程序，如使用 cuDNN 的卷积操作，然后运行程序。如果程序能够正常运行，表示 cuDNN 安装成功。

六、结语

通过本文的指导，你应该已经成功在 Ubuntu 22.04.1 LTS 上安装了 CUDA 11.0 和 cuDNN。现在你可以开始使用这些工具进行深度学习实验了。如果在安装过程中遇到任何问题，可以参考 NVIDIA 官网的文档或寻求社区的帮助。