ubuntu16.04安装CUDA驱动 - 小

背景：项目需要使用PyTorch   ，调用这两个命令nvidia-smi  nvcc --version

　　安装cuda，先安装显卡驱动

检查显卡型号：lspci | grep -i nvidia

01:00.0 VGA compatible controller: NVIDIA Corporation GP106 [GeForce GTX 1060 6GB] (rev a1)
1.确认驱动型号之后:安装依赖 

sudo apt-get update
sudo apt-get install -y gcc g++ make cmake libprotobuf-dev protobuf-compiler
2.使用apt仓库安装 

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo ubuntu-drivers devices # 查看推荐的驱动版本
sudo apt-get install nvidia-driver-450 # 安装推荐的驱动版本，例如450

添加 NVIDIA CUDA仓库：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"
sudo apt-get update

这里有个GPG报错问题：就是系统比较老，不支持https的话，手动下载导入密钥：

sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
这会在你当前的目录下得到一个名为 7fa2af80.pub 的文件
从本地文件中导入密钥：下载完成后，使用 apt-key 命令从这个本地文件导入密钥

sudo apt-key add 7fa2af80.pub  提示ok 就是导入成功

确保完全仓库配置：

sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"

1. 安装CUDA：

仅安装CUDA Toolkit：如果您已经安装了NVIDIA驱动，只需安装CUDA工具包  

sudo apt-get install cuda-toolkit-10.0

安装特定版本的CUDA：

apt-cache search cuda | grep toolkit
例如安装：

sudo apt-get install cuda-toolkit-9-0 （已有驱动前提，安装工具）
sudo apt-get install cuda-10-0 （没有驱动，直接安装）

sudo apt-get install cuda  （最新版本）
安装完成之后：
设置环境变量：

echo 'export PATH=/usr/local/cuda-10.2/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
验证安装：nvcc -V

 

一些前提条件排查可自行百度解决，比如查看有没有系统自带驱动，怎么进行封禁啥的

重启完成之后就可使用了：验证 nvidia-smi