필터 지우기
필터 지우기

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

조회 수: 7 (최근 30일)
ssk
ssk 2019년 3월 6일
댓글: ssk 2019년 3월 9일
以下のリンクを参考にAlexnetの転移学習をGPUで行いたいと考えております。
どのようなGPUの追加コードを加えると動くでしょうか。
どうぞよろしくお願いいたします。

채택된 답변

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 CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

Community Treasure Hunt

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

Start Hunting!