Azure で tensorflow-gpu ~tensorflow-gpu 導入編~

$ sudo apt-get install linux-headers-$(uname -r)

#---- 以下は実行結果です．

Reading package lists... Done
Building dependency tree
Reading state information... Done
linux-headers-4.15.0-1022-azure is already the newest version (4.15.0-1022.23).
linux-headers-4.15.0-1022-azure set to manually installed.
0 upgraded, 0 newly installed, 0 to remove and 26 not upgraded.

Verify the System has the Correct Kernel Headers and Development Packages Installed

そして，以下のコマンドを上から順に実行すると，CUDA のインストールおよびパスの指定が完了します．

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda-9-0
$ export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Install CUDA

CUDA が適切にインストールされているかを確認するため，下記のコマンドを実行します．ここで，CUDA 9.0 がインストールされていることを確認しましょう．

$ nvcc -V

#---- 以下は実行結果です．

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176

Verify the Driver Version

cuDNN をインストール

① 自身のローカルPCで，cuDNN のダウンロードページ (https://developer.nvidia.com/cuda-90-download-archive) に行き，"Download cuDNN >" をクリックします．

② 続いて，"Download cuDNN v7.2.1 (August 7, 2018), for CUDA 9.0" をクリックします．

③ そして，cuDNN v7.2.1 Runtime Library for Ubuntu16.04 (Deb)，cuDNN v7.2.1 Developer Library for Ubuntu16.04 (Deb)，cuDNN v7.2.1 Code Samples and User Guide for Ubuntu16.04 (Deb) をクリックし，ダウンロードを開始します．

以下３つのファイルが自身のローカルPCにダウンロードされたと思います．
libcudnn7_7.2.1.38-1+cuda9.0_amd64.deb
libcudnn7-dev_7.2.1.38-1+cuda9.0_amd64.deb
libcudnn7-doc_7.2.1.38-1+cuda9.0_amd64.deb

自身のローカル PC にダウンロードした libcudnn ファイルたちを，仮想マシンにアップロードする必要があるので，以下の scp コマンドを実行します．

$ scp 先程ダウンロードしたファイルが保存されているディレクトリ/libcudnn* taiki@パブリックIPアドレス:コピー先のディレクトリ

Send libcudnn files to the virtual machine

cd コマンドで，libcunn ファイルがあるディレクトリに移動し，以下のコマンドを実行すると，cuDNN のインストールが完了します．

$ sudo dpkg -i libcudnn7_7.2.1.38-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.2.1.38-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.2.1.38-1+cuda9.0_amd64.deb

Install cuDNN