ubuntu配置cuda环境一:安装NVIDIA驱动

环境:Ubuntu20.04,Gtx1060 6G,Python3.8

1. 安装必要的依赖项

sudo apt-get update	# 更新软件列表
sudo apt-get install g++	# 安装g++,ubuntu自带,可以忽略
sudo apt-get install gcc	# 安装gcc,ubuntu自带,可以忽略
sudo apt-get install make

2. 下载驱动程序

  1. 使用如下命令可查询显卡型号的信息
    lspci | grep -Ei '(vga|display)'
    lspci | grep -i nvidia
    lspci | grep -i vga
    
  2. 官网下载对应的驱动包

    NVIDIA官方下载地址

    选择对应的版本下载即可nvidia

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-C0F6VH0m-1669973021554)(https://pictures-1258408207.cos.ap-nanjing.myqcloud.com/NVIDIA%E9%A9%B1%E5%8A%A8%E4%B8%8B%E8%BD%BD/nvidiadrivcverdownload.png)]

注意:下载得到xxx.run文件,存放路径中不能有中文,记下下载位置备用

  1. 卸载原有驱动

    通过如下命令卸载原有的驱动,再重新安装新驱动:

    sudo apt-get remove --purge nvidia*
    
  2. 禁用nouveau

    在安装nvidia驱动前需先卸载ubuntu自带的显卡驱动nouveau:

    sudo gedit /etc/modprobe.d/blacklist.conf	# 打开配置文件
    # 在该文件的结尾加上两行
    blacklist nouveau
    options nouveau modeset=0
    # 保存文件关闭gedit
    
    sudo update-initramfs -u	# 使配置文件生效
    
    # 重启系统,重启后由于禁用了显卡驱动,故不能进入图形界面,Ctrl + Alt + F2进入命令行
    sudo reboot
    
    # 查看nouveau是否被禁用
    lsmod | grep nouveau	# 若没有输出,则表明nouveau禁用成功
    
    1. 安装驱动

      为了安装Nvidia驱动程序,首先得禁用当前的显示服务器,最简单的方法就是使用telinit命令将其运行级别改为3

      sudo telinit 3
      

      禁用X-Window服务

      sudo /etc/init.d/gdm3 stop
      

      进入存放驱动文件的目录,更改权限

      sudo chmod 777 NVIDIA-Linux-x86_64-515.65.01.run   #给你下载的驱动赋予可执行权限,才可以安装
      sudo ./NVIDIA-Linux-x86_64-515.65.01.run –no-opengl-files   #安装
      

      安装完成后重启相关服务

      sudo service gdm3 start	# 自动进入图形界面
      

      可以使用如下命令查看是否安装成功

      nvidia-smi
      

      [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-gvNMqyrv-1669973021560)(https://pictures-1258408207.cos.ap-nanjing.myqcloud.com/NVIDIA%E9%A9%B1%E5%8A%A8%E4%B8%8B%E8%BD%BD/nvidia-smi.png)]

或者

nvidia-settings	# 调出nvidia驱动设置界面

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SENcHdXo-1669973021561)(https://pictures-1258408207.cos.ap-nanjing.myqcloud.com/NVIDIA%E9%A9%B1%E5%8A%A8%E4%B8%8B%E8%BD%BD/nvidia-settings.png)]

你可能感兴趣的:(ubuntu)