久しぶりにCNTKを触ってみたら、以前の記事とインストール手順が大分変わっていたのでメモ。以前は必要なランタイムのインストールは手動で行うしかありませんでしたが、インストールスクリプトで自動的に行ってくれるようになっています。
対象OSとなるWindowsは64bit版のみです。CUDAに対応していますが、CUDAに対応したGPGPUを搭載していなくても動作します。
CNTKを動かすにはCUDA7.0が必要になるので、NVIDIAのホームページからCUDA 7.0 (https://developer.nvidia.com/cuda-toolkit-70)をダウンロードしてインストールします。
CNTKのバイナリをhttps://github.com/Microsoft/CNTK/releasesからダウンロードします。GPU版とCPU版が用意されていますが、私が試したときにはGPU版しか置かれていませんでした。GPGPUが無い環境でもCUDA7.0をインストールしていれば、GPU版を動作させることができます。
ダウンロードしたバイナリをC:\local\cntkに展開します。
Powershellを管理者権限で起動して以下のコマンドを実行します。これによりAnaconda3のインストール他、必要な環境の設定をおこなってくれます。
cd c:\local\cntk\Scripts\install\windows .\install.ps1 -execute
最初に何度かセキュリティ警告が表示されますが、全て「[R]一度だけ実行する」を選択してください。
途中で以下のように選択が表示されるので1を入力してください。
1 - I agree and want to continue Q - Quit the installation process
最後に以下のように確認が表示されるのでyを入力してください。
Do you want to continue? (y/n)
以上でインストール完了します。
サンプルを実行するにはコマンドプロンプトを開いて、最初に以下のコマンドを実行します。
c:\local\cntk\scripts\cntkpy34.bat