CUDAを使用しないGPUを用いた深層学習の実行

現在GPU（GeForce 1080-Ti）を用いたLSTMによる深層学習を行っています。

これまではMatlabのバージョンのアップデートに伴い、対応するCUDA等のインストールを検討していたのですが、調査する中で「Parallel Computing Toolbox に実装されている関数を使用する場合はCUDAのインストールは不要」との記載を見つけました。

深層学習の学習過程でGPUを使用しているので、CUDAやcudnnも併せて使用していると考えていたのですが、Deep Leaning Toolboxに実装されているモデルを用いて学習を行う場合は、CUDA等はインストールしなくてもGPUで実行出来るのでしょうか。

Answer 1

[Kojiro Saito]

はい、Deep Learning ToolboxのtrainNetworkでモデルの学習をGPUで行う場合は、CUDA等のインストールは不要です。

こちらの表にまとまっていますが、CUDAカーネルオブジェクトを CUコードから生成する場合と、GPU CoderでCUDAのソースコードにコンパイルする場合には、CUDA ツールキットのインストールが必要です。

GPU での CUDA または PTX コードの実行

[中尾 圭吾]

回答頂きありがとうございます。

Deep Learning Toolboxのみを用いたモデルの学習の場合は、CUDAのインストールが必要ないとの旨承知しました。

最新版にバージョンを更新したいと思います。