Linux安装显卡驱动和CUDA

一、安装显卡驱动

方法一

1.查询系统中是否安装了显卡驱动

命令行键入：

nvidia-smi

下图是未安装的显示

 

2.查询显卡型号并选择安装的驱动版本 

（1）查询显卡型号

命令行键入：

lspci |grep -i vga

（2）根据显卡型号选择合适的显卡驱动

这里下载：https://www.nvidia.com/Download/index.aspx

 点击下载

 

 

3.安装驱动

命令行键入：

sh .run文件存放的路径

然后跟着说明一步一步安装就行了。

安装成功后键入 nvidia-smi 命令，看是否安装成功。

4.方法二——在线安装

键入 ubuntu-drivers devices 查看推荐安装的驱动版本

 

sudo apt-get install nvidia-driver-535
# 535 为驱动的版本号，可以在 ubuntu-drivers devices 中查看推荐版本

点击回车

 

 

reboot # 重启

5.方法三——在线自动安装（推荐）

1.   #自动化安装显卡驱动
2.   ubuntu-drivers devices
3.   sudo ubuntu-drivers autoinstall
4.   sudo reboot

三、安装CUDA

1.查看系统中是否有CUDA

命令行键入：

nvcc -V  或  nvcc --version

显示如下界面就说明没有CUDA

 

2.选择合适的CUDA版本

（1）命令行键入 nvidia-smi 可以看到系统支持的最大CUDA版本。

 

3.CUDA下载及安装

点击下载CUDA

你会进入到如下界面，选择你需要的CUDA版本

 

 

然后在命令行中依次键入以下命令安装

第一行是下载安装文件 .run

第二行是安装

 

然后在命令行中依次键入以下命令安装

第一行是下载安装文件 .run

第二行是安装

 

然后在命令行中依次键入以下命令安装

第一行是下载安装文件 .run

第二行是安装

 

 

安装中。。。。

进入漫长的等待。。。。。。

键入accept

 

如果已经安装显卡驱动，则取消掉“Driver”这个选项（在Driver位置键入 enter取消）

光标移至Install，按ENTER键

 

漫长的等待。。。。。

然后显示下面内容就安装成功了

 

4.配置环境变量 

vim ~/.bashrc

将下面两行代码写进 .bashrc 文件中。

怎么写呢？使用vim 打开这个文件后，将下面两行复制进去，然后光标移动到文件末尾，键入:wq即可 。

 

#在安装成功后的显示中可以看到cuda的安装位置
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然后再 source ~/.bashrc 一下，更新一下变量。

然后命令行中键入 nvcc -V 看一下是否安装成功

显示下面内容就是安装成功啦~~