当前位置:首页 > 系统教程 > 正文

Ubuntu安装NVIDIA显卡驱动、CUDA与CuDNN完整指南(从零开始,小白也能轻松搞定)

Ubuntu安装NVIDIA显卡驱动、CUDA与CuDNN完整指南(从零开始,小白也能轻松搞定)

在本教程中,我们将详细介绍如何在Ubuntu系统上安装NVIDIA显卡驱动、CUDA工具包以及CuDNN库。这是进行深度学习和GPU加速计算的基础步骤。即使你是初学者,也能按照步骤轻松完成。

Ubuntu安装NVIDIA显卡驱动、CUDA与CuDNN完整指南(从零开始,小白也能轻松搞定) NVIDIA显卡驱动安装  CUDA工具包安装 CuDNN库配置 Ubuntu系统教程 第1张

准备工作

在开始安装之前,请确保你的系统是Ubuntu,并且有NVIDIA显卡。首先,更新系统包列表:sudo apt updatesudo apt upgrade。这有助于确保系统兼容性。

步骤一:安装NVIDIA显卡驱动

安装NVIDIA驱动是第一步。你可以通过Ubuntu的附加驱动或命令行安装。这里我们使用命令行方法。

首先,添加NVIDIA驱动PPA:sudo add-apt-repository ppa:graphics-drivers/ppa,然后更新:sudo apt update

接下来,安装推荐的NVIDIA驱动。使用命令 ubuntu-drivers devices 查看可用驱动,然后安装:sudo apt install nvidia-driver-XXX(XXX为版本号)。重启系统使驱动生效。这个过程是NVIDIA显卡驱动安装的关键。

步骤二:安装CUDA工具包

CUDA工具包是NVIDIA提供的并行计算平台。访问NVIDIA官网下载对应版本的CUDA工具包。这里以CUDA 11.8为例。

下载后,运行安装命令:sudo sh cuda_11.8.0_520.61.05_linux.run。按照提示安装,注意在安装选项中选择不安装驱动(如果已安装驱动)。

安装完成后,添加环境变量。编辑 ~/.bashrc 文件,添加:export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}。然后执行 source ~/.bashrc。这完成了CUDA工具包安装。

步骤三:安装CuDNN库

CuDNN是深度神经网络库,需要从NVIDIA开发者网站下载。注册账号后,下载对应CUDA版本的CuDNN。

下载后,解压文件并复制到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*。这实现了CuDNN库配置。

验证安装

验证NVIDIA驱动:nvidia-smi。验证CUDA:nvcc --version。验证CuDNN:检查头文件或运行示例。

完成以上步骤后,你的Ubuntu系统就成功配置了NVIDIA显卡驱动、CUDA工具包和CuDNN库,可以开始深度学习项目了。这个Ubuntu系统教程旨在帮助新手快速上手。

本教程涵盖了从NVIDIA驱动安装到CuDNN配置的全过程,希望对你有所帮助。如果你遇到问题,可以参考官方文档或社区论坛。