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 ，請轉到此處： ****

參考：