CUDA環境をセットアップしてSonyのNeural Network Consoleを動かすまで
YouTubeでSonyの小林さんという方が公開されているDeep Learning基礎の動画を見て何となく触発されてしまったので、3ヶ月ほど放置状態だった我が家のGTX 1080でDeep Learningを動かしてみることにしました。
環境
- Windows 10 Pro 1903 (64-bit)
- ELSA GeForce GTX 1080 8GB S.A.C
- NVIDIA グラフィックス ドライバー 441.66
- CUDA Toolkit 10.2
- NVIDIA cuDNN 7.6.5
- Neural Network Console 1.6
GPUドライバとかCUDAとかダウンロード
「何を今更」と思うかもしれませんが私はここでドハマりしたので念の為最新版をダウンロードしておきます。
GPUドライバは該当ファミリーの「Studio ドライバー (SD)」でいいです。ゲームやる人は「Game Ready Driver (GRD)」を選ぶとよいかもです。
cuDNNはNVIDIAのアカウントが必要です。
Neural Network Consoleダウンロード
Windows版をダウンロードします。公式から順当に進むと危うくクラウド版に進んでしまいそうになるのでご注意。
今どきインストーラーなしなので適当な場所に解凍します。
CUDA設定
Neural Network ConsoleでCUDAを使うように設定します。
起動したら右上のカバンみたいなボタンからSetup→ENGINEを開き、Processor TypeをCUDA-Enabled GPUに切り替えます。
いざ、動かす
あとは何も面倒なことはなく、サンプルのプロジェクトを動かしてみます。
今回はとりあえず一番シンプルそうな「01_logistic_regression」を選びました。
ね、簡単でしょう？
