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

轻云UC

已于 2024-04-28 08:25:44 修改

阅读量253

点赞数 4

分类专栏： AI技术文章标签： linux 环境

于 2024-04-26 12:31:38 首次发布

本文链接：https://blog.csdn.net/tanlintanlin/article/details/138215094

版权

AI技术专栏收录该内容

4 篇文章 0 订阅

订阅专栏

一、安装准备

安装依赖

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