cuda和cudnn的安装(ubuntu22.04环境)

一、安装准备

安装依赖

sudo apt-get update
sudo apt-get install g++
sudo apt-get install gcc
sudo apt-get install make

禁用默认驱动

sudo gedit /etc/modprobe.d/blacklist.conf
在末尾加上

blacklist nouveau
options nouveau modeset=0

更新一下initramfs -u的镜像

sudo update-initramfs -u

完成以上步骤后，重启电脑。然后在终端中输入检查：

lsmod | grep nouveau

二、NVIDIA驱动安装

配置安装

选择一个你要安装的驱动，点击应用更改，安装完成后直接重启，完成安装。

检查：

执行nvidia-smi 命令，看到如下界面：

三、cuda安装

下载并安装cuda

wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.runsudo sh cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run

• 会看到如下界面-> 选择continue， 在弹出界面中，输入 accept 即可下一步

• 选择要装的部内容，直接 定位到最下面的Install即可进行安装

修改环境变量：

sudo vim ~/.bashrc

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

source ~/.bashrc

四、安装cuDNN

下载并安装cudnn

wget https://developer.download.nvidia.com/compute/cudnn/9.1.0/local_installers/cudnn-local-repo-ubuntu2204-9.1.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.0_1.0-1_amd64.deb

进入到如下目录监测相关文件是否存在：

拷贝文件并更新

sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update

安装cudnn-cuda-12

sudo apt-get -y install cudnn-cuda-12

查看版本并检查

nvcc -V