【Ubuntu20.04安装Nvidia驱动、CUDA和CUDNN】

1 Nvidia驱动安装

官网链接:https://www.nvidia.cn/Download/index.aspx?lang=cn
或者https://www.nvidia.cn/geforce/drivers/

:Ubuntu系统是不区别显卡类别的显卡驱动,一般来说,下载最新版本的驱动即可;Win系统是需要根据显卡来选择具体的驱动版本。

1.1 安装

# 安装
sh ./NVIDIA-Linux-x86_64-535.98.run
# 验证安装是否成功
nvidia-smi

在这里插入图片描述

1.2 安装Nvidia可能会遇到的问题

1.2.1 NVIDIA 驱动与 Nouveau 驱动不兼容

由于系统当前正在使用 Nouveau 显卡驱动导致的,而 NVIDIA 驱动与 Nouveau 驱动不兼容。为了解决这个问题,需要禁用 Nouveau 驱动。
在这里插入图片描述
在这里插入图片描述

解决方案:
1、打开终端并切换到 root 用户。
编辑 /etc/modprobe.d/blacklist.conf 文件,在文件末尾添加以下内容:
blacklist nouveau options nouveau modeset=0
保存并关闭该文件。

2、重新生成 initramfs:
sudo update-initramfs -u

3、重启系统
reboot

1.2.2 ERROR: Unable to find the development tool cc

在这里插入图片描述
在这里插入图片描述

sudo apt update
sudo apt install build-essential
sudo apt-get install manpages-dev

2 CUDA安装

2.1 下载和安装

CUDA官网:https://developer.nvidia.com/cuda-toolkit-archive

# 下载
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
# 安装
sudo sh cuda_12.2.0_535.54.03_linux.run

注:需要取消驱动安装(Driver)
在这里插入图片描述

2.2 配置CUDA环境

打开 ~/.bashrc文件

sudo gedit ~/.bashrc

在文件中的最后加入

export PATH="/usr/local/cuda/bin/:$PATH"
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export CUDA_HOME=/usr/local/cuda

刷新环境

source ~/.bashrc

3 安装CUDNN

CUDNN下载官网:https://developer.nvidia.com/rdp/cudnn-archive
,选择与CUDA版本对应的即可。
下载得到一个压缩包,解压之后将里面的文件各自复制到目标文件夹下,可参考执行如下命令:

# 复制cudnn头文件
sudo cp cuda/include/* /usr/local/cuda-12.2/include/
# 复制cudnn的库
sudo cp cuda/lib64/* /usr/local/cuda-12.2/lib64/
# 添加可执行权限
sudo chmod +x /usr/local/cuda-12.2/include/cudnn.h
sudo chmod +x /usr/local/cuda-12.2/lib64/libcudnn*

4 切换CUDA版本

使用的cuda是/usr/local/cuda这个软连接,所以更换CUDA版本,删除旧cuda软连接文件夹,然后新建一个别的版本cuda软连接到新的cuda软连接文件夹即可。

#删除旧cuda软连接
sudo rm -rf /usr/local/cuda
#新建新的cuda版本软连接
sudo ln -s /usr/local/cuda-11.7 /usr/local/cuda   
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值