2019年4月15日 更新

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

2,920 view お気に入り 0
$ 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 をインストール

 (4972)

① 自身のローカル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) をクリックし,ダウンロードを開始します.

以下3つのファイルが自身のローカル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
.

tensorflow-gpu をインストール


最後に,以下のコマンドを実行することで,tensorflow-gpu のインストールが完了します.
33 件

関連する記事 こんな記事も人気です♪

Python × TensorFlow ② ~TensorFlow を扱う上で必要な知識「定数・変数」~

Python × TensorFlow ② ~TensorFlow を扱う上で必要な知識「定数・変数」~

TensorFlow を扱う上で必要な以下3つの知識のうち「定数と変数」について解説していきたいと思います.
井上 大輝 | 7,713 view
Azure + Tensorflow + Kerasで画像認識

Azure + Tensorflow + Kerasで画像認識

クラウドでGPUを使ったディープラーニングを実行するための環境構築法をまとめました。
山田涼太 | 4,974 view

この記事のキーワード

この記事のキュレーター

井上 大輝 井上 大輝