Ubuntu22.04+cuda11.7+cudnn8.4.1安装

happyitdad

已于 2022-09-22 15:04:56 修改

阅读量3.4k

点赞数 1

文章标签： linux 服务器 ubuntu

于 2017-10-17 16:43:06 首次发布

本文链接：https://blog.csdn.net/u012351661/article/details/78262214

版权

1、安装nvidia驱动

1.1 卸载原有驱动

sudo apt-get purge nvidia*

1.2 禁用nouveau

创建blacklist-nouveau.conf

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

内容写上：

blacklist nouveau
options nouveau modeset=0

更新initramfs

sudo update-initramfs -u

重启机器，重新进入系统后，检验nouveau是否被禁用

lsmod | grep nouveau

如果没有显示内容，则表示nouveau被成功禁用。

1.3 下载驱动

到 https://www.nvidia.com/zh-cn/ 根据自己的显卡型号下载.run文件，我下载的是：NVIDIA-Linux-x86_64-515.57.run。

1.4 安装驱动

结束X-window服务

sudo service lightdm stop

执行.run文件

sudo apt install pkg-config libglvnd-dev
sudo sh ./NVIDIA-Linux-x86_64-515.57.run

提示成功后，重启系统，然后可以执行 nvidia-smi 检查驱动是否正常。

2、安装CUDA

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

因为驱动之前已经安装，这里就不要选择安装驱动。其余的都直接默认或者选择是即可。

修改/etc/profile，添加以下内容：

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

执行 . /etc/profile 使环境变量生效。

3、测试cuda的Samples

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery

如果显示的是一些关于GPU的信息，则说明安装成功了。

4、安装cudnn8.4.1

tar xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
cd cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/include
sudo cp include/* /usr/local/cuda/include
sudo cp lib/* /usr/local/cuda/lib64