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

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
  保存后,重启机器。

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

你可能感兴趣的:(linux,服务器,ubuntu)