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

1 次查看（过去 30 天）
ssk
ssk 2019-3-6
评论： ssk 2019-3-9
以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。
どのようなGPUの追加コードを加えると動くでしょうか。
https://jp.mathworks.com/help/deeplearning/examples/transfer-learning-using-alexnet.html
どうぞよろしくお願いいたします。

请先登录，再回答此问题。

采纳的回答

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

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!