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

Question

中尾圭吾 2022-11-18

评论：中尾圭吾 2022-11-19

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

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

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

Answer 1

Kojiro Saito 2022-11-18

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

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

中尾圭吾 2022-11-19

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

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

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