Ubuntu 安裝顯卡驅動 +cuda +cudnn

1、安裝顯卡驅動

ubuntu安裝顯卡驅動的三種方法

第三種方法可行：然後在“附件驅動”中找到對應版本重啓即可。
但是需要只有對應的一個內核版本

第四種方法: 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())