Ubuntu 安装显卡驱动（四种方法） +cuda +cudnn

大家好，欢迎来到IT知识分享网。

1、安装显卡驱动

参考：ubuntu安装显卡驱动的三种方法

第三种方法成功：

• sudo add-apt-repository ppa:graphics-drivers/ppa
• sudo apt-get update
• sudo apt-get install nvidia-418 nvidia-settings nvidia-prime
• nvidia-settings
• 在“附件驱动”中找到对应版本重启即可

第四种方法:

一句即可：sudo ubuntu-drivers autoinstall

出现问题

安装好后还是显示NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

链接中第一个方法解决：输入sudo modprobe nvidia

2、安装cuda+cudnn

参考：Ubuntu 16.04 上 CUDA_10.0及cuDNN的安装
cuda安装:Ubuntu16.04：NVIDIA cuda10.0安装
验证：Ubuntu18.04+英伟达驱动+cuda10.0+cudnn7.6.3+tensorflow-gpu

• 下载cuda安装包：cuda官网下载，，根据系统信息选择对应的版本，runfile安装的话最后一项要选择 runfile文件，如下图片。
  
• MD5码验证：md5sum cuda_10.0.130_410.48_linux.run
• 安装：sudo sh cuda_10.0.130_410.48_linux.run
  
• 查看是否安装成功:ls /dev/nvidia*
  
• 设置环境变量：sudo gedit ~/.bashrc
  添加以下内容：

export PATH=$PATH:/usr/local/cuda/bin

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/bin/lib64:$LD_LIBRARY_PATH

• source ~/.bashrc

• cd NVIDIA_CUDA-10.0_Samples/

• make
  

• cd NVIDIA_CUDA-10.0_Samples/bin/x86_64/linux/release

• ./deviceQuery
  

• cudnn下载：
  

• 下载完成并解压后得到一个cuda文件夹后，执行以下命令：

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

• 利用torch测试cuda是否可用：

import torch
print(torch.cuda.is_available())