ubuntu篇---ubuntu20.04安装cuda和cudnn

心惠天意

已于 2024-07-09 10:34:41 修改

阅读量1.7w

点赞数 11

分类专栏： ubuntu 文章标签： ubuntu

于 2021-11-19 11:17:36 首次发布

本文链接：https://blog.csdn.net/m0_46825740/article/details/121418113

版权

ubuntu 专栏收录该内容

6 篇文章

订阅专栏

ubuntu篇---ubuntu20.04安装cuda和cudnn

1. 前期准备
2. 安装显卡驱动
3. 安装cuda
4. 安装cuDNN
5. 安装N卡驱动（建议不要用ubuntu自带的显卡驱动）

操作系统环境：Ubuntu 20.04

1. 前期准备

1.1 先卸载所有的驱动

sudo apt-get remove nvidia*

在这里插入图片描述

1.2 更新软件列表和安装必要软件、依赖

终端输入以下命令：

sudo apt-get update
     
sudo apt-get install g++
     
sudo apt-get install gcc
    
sudo apt-get install make

1.3 查询硬件(显卡)信息

lspci | grep -Ei '(vga|display)‘ 或者 lspci | grep -i nvidia 或者 lspci | grep -i vga

在这里插入图片描述

1.4 官网下载对应驱动

Nvidia官网
 Nvidia驱动下载地址
 cuda下载
进入如下界面，选择对应的驱动程序：
在这里插入图片描述

下载后得到xxx.run文件(如：NVIDIA-Linux-x86_64-530.41.03.run)，注意存放路径不要有中文

1.5 禁用默认驱动(nouveau是通用的驱动程序)

在安装NVIDIA驱动以前需要禁止系统自带显卡驱动nouveau。

在终端输入命令打开blacklist.conf文件。

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

或者新建一个单独的blacklist-nouveau.conf文件。

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

如果没有Vim，根据提示安装Vim。

sudo apt-get install vim

在打开的文件末尾输入并保存：

blacklist nouveau
options nouveau modeset=0

ESC退出后，再输入 :wq 进行修改保存

最后更新一下系统的initramfs镜像文件，在终端中输入：

sudo update-initramfs -u

完成以上步骤后，重启电脑。然后在终端中输入：

lsmod | grep nouveau

在这里插入图片描述

如果没有输出结果，表示禁用成功，没有成功则不能进入下一步。

2. 安装显卡驱动

安装 1.4 步骤 下载的显卡驱动

sudo chmod 777 NVIDIA-Linux-x86_64-530.41.03.run   # 给下载的驱动赋予可执行权限
     
sudo ./NVIDIA-Linux-x86_64-530.41.03.run -no-opengl-files -no-x-check -no-nouveau-check  # 安装
# 若安装错误，可以卸载驱动
sudo ./NVIDIA-Linux-x86_64-530.41.03.run --uninstall

启动安装程序，可以加参数（安装驱动前需要先安装gcc和make）。

我在各种教程里面参数有：
–no-opengl-files：表示只安装驱动文件，不安装OpenGL文件。这个参数不可省略，否则会- 导致登陆界面死循环，英语一般称为”login loop”或者”stuck in login”。然而我并没有用
–no-x-check：表示安装驱动时不检查X服务。如果没有这个参数，可能会出现“X-Server needs to be disabled before installing the drivers”的错误。然而我也没有用
–no-nouveau-check：表示安装驱动时不检查nouveau，非必需。
-Z, --disable-nouveau：禁用nouveau。此参数非必需，因为之前已经手动禁用了nouveau。
-A：查看更多高级选项。

以下是我安装的时候遇到的问题，可供参考

在这里插入图片描述

成功安装。。。。
在这里插入图片描述

百度搜索到的问题（仅供参考）

1.The distribution-provided pre-install script failed! Are you sure you want to continue?
选择continue installation

2.Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?  
选择 No 继续。

3.问题没记住，
选项是：install without signing

4.问题大概是：Nvidia's 32-bit compatibility libraries? 
选择 No 继续。

5.Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  
选择 Yes  继续

我的具体代码执行如下
在这里插入图片描述验证 nvidia-smi

在这里插入图片描述

3. 安装cuda

进入nvidia开发者网站的CUDA下载页面：
CUDA Toolkit Archive
选择deb(local)格式的CUDA文件下载

在这里插入图片描述安装步骤同官网

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb

sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

然后

sudo vim ~/.bashrc

我们在文件最后一行添加：

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

进入/usr/local/cuda-11.2/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

在这里插入图片描述

4. 安装cuDNN

NVIDIA cuDNN
选择合适的版本下载（cuDNN Library for Linux）
在这里插入图片描述然后解压，
解压命令为：

tar zxvf cudnn-11.2-linux-x64-v8.1.1.33.tgz -C .

并进入到相应目录，运行以下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64
sudo chmod a+r /usr/local/cuda-11.2/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-11.2/lib64/libcudnn*

以配置cuDNN环境
以上则配置完成！

另一种方式
在这里插入图片描述

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux.run

sudo sh cuda_11.4.2_470.57.02_linux.run

不要选择显卡驱动

在这里插入图片描述

下载cuda

附加：

5. 安装N卡驱动（建议不要用ubuntu自带的显卡驱动）

首先我们需要添加源，

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

然后检查可以安装的驱动版本，

ubuntu-drivers devices

在这里插入图片描述

选择最合适的版本安装即可，

sudo apt install nvidia-driver-XXX

这里的xxx是驱动的版本，可以百度查一下。

安装成功后，查看是否安装成功

nvidia-smi

如果没成功，则先reboot重启系统，则可以发现进入系统再输入查看驱动命令就成功了。

参考链接 https://blog.csdn.net/To_be_little/article/details/123630010