标签搜索

目 录CONTENT

文章目录

【环境配置】Linux GPU 环境配置(驱动、Cuda和Cudnn安装)

览平科技
2022-08-04 / 0 评论 / 0 点赞 / 1,353 阅读 / 658 字 / 正在检测是否收录...
温馨提示:
本文最后更新于 2022-08-11,若内容或图片失效,请留言反馈。部分素材来自网络,若不小心影响到您的利益,请联系我们删除。

Linux GPU 环境配置(驱动、Cuda和Cudnn安装)

注意,由于Nvidia 驱动、Cuda和Cudnn的版本实时更新,因此本文中涉及版本号的均由‘x’代替!!!

驱动安装

驱动下载

点击链接,根据显卡型号、操作系统下载相应驱动 Nvidia Driver Download Page

禁用 nouveau

打开终端,输入以下:

sudo gedit /etc/modprobe.d/blacklist.conf 

在 blacklist.conf 文件末尾添加并保存:

blacklist nouveau

然后执行命令更新:

sudo update-initramfs -u

重启并验证是否禁用了 nouveau:

lsmod | grep nouveau

无任何信息返回,则已被禁用。

驱动安装

卸载已安装的 Nvidia 驱动:

sudo apt-get remove --purge nvidia*

执行安装文件:

sudo bash ./NVIDIA-Linux-x86_64-xxx.xx.run -no-x-check -no-nouveau-check -no-opengl-files

验证是否安装成功:

nvidia-smi

Cuda 安装

Cuda下载

点击链接,根据系统类型,系统版本下载相应文件 Nvidia Cuda Download Page

依次点击:“Linux” -> “x86_64” -> “Ubuntu” -> “22.04” -> “runfile (local)”

这里作者建议采用"runfile (local)"的方式,简单快捷,不易出错。

此时,下载页面会显示如下执行信息:

wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
sudo sh cuda_11.7.1_515.65.01_linux.run

Cuda安装

在安装时,切记取消选择驱动安装!!! 因上述步骤中,已经安装了最新版本的驱动。

执行完上述文件后,需将cuda中的bin目录添加到Path里,lib目录添加到LD_LIBRARY_PATH里。

打开环境变量配置文件:

vim ~/.bashrc

添加环境变量到文件底部:

export PATH=/usr/local/cuda-11.x/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.x/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并退出,验证Cuda是否安装成功:

nvcc -V

Cudnn安装

Cudnn下载

点击链接,根据Cuda版本下载相应文件 Nvidia Cudnn Download Page

这里,作者建议下载 “Local Installer for Linux x86_64 (Tar)”类型文件。

Cudnn安装

解压tar文件:

tar -xvf cudnn-linux-xxx-archive.tar

进入解压后的文件夹,并将lib下所有文件复制到cuda lib64文件夹下,include下所有文件复制到cuda include文件夹下:

cd cudnn
sudo cp lib/* /usr/local/cuda-11.x/lib64/
sudo cp include/* /usr/local/cuda-11.x/include/

查看并验证Cudnn版本信息:

cat /usr/local/cuda-11.x/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
0

评论区