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版本对应的即可。
下载得到一个压缩包,解压之后将里面的文件各自复制到目标文件夹下,可参考执行如下命令:文章来源:https://www.toymoban.com/news/detail-672550.html
# 复制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软连接文件夹即可。文章来源地址https://www.toymoban.com/news/detail-672550.html
#删除旧cuda软连接
sudo rm -rf /usr/local/cuda
#新建新的cuda版本软连接
sudo ln -s /usr/local/cuda-11.7 /usr/local/cuda
到了这里,关于【Ubuntu20.04安装Nvidia驱动、CUDA和CUDNN】的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!