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

第二行是安装

 

             wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.runsudo sh cuda_12.2.0_535.54.03_linux.run

    sudo sh cuda_12.2.0_535.54.03_linux.run

安装中。。。。

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

键入accept

 

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

光标移至Install,按ENTER键

 

漫长的等待。。。。。

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

 

4.配置环境变量 

vim ~/.bashrc

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

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

 

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

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

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

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

 

 

posted @ 2024-03-08 16:00  zhihua  阅读(1216)  评论(0编辑  收藏  举报