Azure N 系列（GPU）在 UBUNTU 16.04 LTS 上安装 CUDA cudnn Tensorflow

Created: November-22, 2018

花了 5 个多小时后，我找到了这个简单的解决方案：

- 要验证系统是否具有支持 CUDA 的 GPU，请运行以下命令：

lspci | grep -i NVIDIA

你将看到类似于以下示例的输出（显示 NVIDIA Tesla K80 / M60 卡）：

StackOverflow 文档

- 禁用 nouveau 驱动程序：

sudo -i
rmmod nouveau

- 重启后 ：sudo reboot，通过发出以下命令验证驱动程序是否安装正确：

lsmod | grep -i nvidia

- 接下来，从 Nvidia 下载 CUDA 包，……

wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb

-…告知 apt-get 并安装 CUDA Toolkit：

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
sudo apt-get update
sudo apt-get install -y cuda

- 现在我们可以通过运行来检查 GPU 的状态：

nvidia-smi

接下来，我们下载 cuDNN …

wget http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz

-…解压缩，复制 lib64 并包含文件夹：

tar -zxf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/
sudo rm -R cuda

- 时间做一些清理并删除下载的档案：

rm cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
rm cudnn-8.0-linux-x64-v5.1.tgz

要使用 CPU / GPU 安装 Tensorflow ，请转到此处： ****

参考：