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

22 views (last 30 days)
中尾 圭吾
中尾 圭吾 on 18 Nov 2022
Commented: 中尾 圭吾 on 19 Nov 2022
現在GPU(GeForce 1080-Ti)を用いたLSTMによる深層学習を行っています。
これまではMatlabのバージョンのアップデートに伴い、対応するCUDA等のインストールを検討していたのですが、調査する中で「Parallel Computing Toolbox に実装されている関数を使用する場合はCUDAのインストールは不要」との記載を見つけました。
深層学習の学習過程でGPUを使用しているので、CUDAやcudnnも併せて使用していると考えていたのですが、Deep Leaning Toolboxに実装されているモデルを用いて学習を行う場合は、CUDA等はインストールしなくてもGPUで実行出来るのでしょうか。

Accepted Answer

Kojiro Saito
Kojiro Saito on 18 Nov 2022
はい、Deep Learning ToolboxのtrainNetworkでモデルの学習をGPUで行う場合は、CUDA等のインストールは不要です。
こちらの表にまとまっていますが、CUDAカーネルオブジェクトを CUコードから生成する場合と、GPU CoderでCUDAのソースコードにコンパイルする場合には、CUDA ツールキットのインストールが必要です。
  1 Comment
中尾 圭吾
中尾 圭吾 on 19 Nov 2022
回答頂きありがとうございます。
Deep Learning Toolboxのみを用いたモデルの学習の場合は、CUDAのインストールが必要ないとの旨承知しました。
最新版にバージョンを更新したいと思います。

Sign in to comment.

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!