MATLAB Answers

Alexnetの転移学習をGPU環境で行うには

2 views (last 30 days)
ssk
ssk on 6 Mar 2019
Commented: ssk on 9 Mar 2019
以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。
どのようなGPUの追加コードを加えると動くでしょうか。
どうぞよろしくお願いいたします。

  0 Comments

Sign in to comment.

Accepted Answer

mizu
mizu on 9 Mar 2019
ドキュメント内の trainNetwork 関数でニューラルネットワークのパラメタ学習を行っています。trainNetwork 関数は、デフォルトで GPU を使用する仕様になっています。そのため、GPU を使用するためのコードを追加する必要はありません。CPU を使用したい場合は 'ExecutionEnvironment' オプションを 'cpu' にします。
もし GPU のあるマシンで CPU でしか実行されていない場合は、gpuDevice を使用して、MATLAB が GPU を認識できているかを確認することができます。

  1 Comment

ssk
ssk on 9 Mar 2019
デフォルトでGPUを使用する仕様となっている旨、ご教示いただきありがとうございました。

Sign in to comment.

More Answers (0)