gpuDeviceCount

存在する GPU デバイスの数

ページ内をすべて折りたたむ

    構文

    n = gpuDeviceCount
    n = gpuDeviceCount(countMode)
    [n,indx] = gpuDeviceCount(___)

    説明

    n = gpuDeviceCount は、GPU デバイス ドライバーによって報告された、ローカル マシンに存在する GPU デバイスの数を返します。MATLAB® でサポートされていないデバイスや、現在の MATLAB セッションで使用できないデバイスも含めて、ドライバーによって報告されたすべてのデバイスがカウントされます。

    例

    n = gpuDeviceCount(countMode) は、countMode に従ってカウントされた、マシン内の GPU デバイスの数を返します。サポートされている GPU デバイスのみをカウントする場合や、この MATLAB セッションで使用可能なデバイスのみをカウントする場合は、この構文を使用します。

    例

    [n,indx] = gpuDeviceCount(___) は、前出の構文のいずれかに関して、カウントされた GPU デバイスのインデックスも返します。カウントされた GPU デバイスを選択または調査する場合は、この構文を使用します。

    例

    例

    すべて折りたたむ

    コンピューター内の使用可能な GPU デバイスの数と、それらのインデックスを特定します。

    [n,indx] = gpuDeviceCount
    n = 2
indx =   
     1     2

    gpuDeviceTable を使用して GPU のプロパティをクエリします。

    gpuDeviceTable
    ans = 
    Index        Name              ComputeCapability    DeviceAvailable    DeviceSelected
    _____    __________________    _________________    _______________    ______________

      1      "GeForce RTX 4090"          "8.9"               true              true     
      2      "GeForce RTX 4090"          "8.9"               true              false
    ライブ スクリプトを開く

    複数の GPU にアクセスできる場合、並列プールを使用して複数の GPU で並列計算を実行することができます。

    MATLAB で使用可能な GPU の数を特定するには、関数gpuDeviceCountを使用します。 

    availableGPUs = gpuDeviceCount("available")
    availableGPUs = 
3

    使用可能な GPU と同数のワーカーをもつ並列プールを起動します。最良のパフォーマンスを得るために、MATLAB は既定でワーカーごとに別々の GPU を割り当てます。

    parpool("Processes",availableGPUs);
    Starting parallel pool (parpool) using the 'Processes' profile ...
Connected to the parallel pool (number of workers: 3).

    各ワーカーが使用している GPU を識別するために、spmd ブロック内で gpuDevice を呼び出します。spmd ブロックは各ワーカーで gpuDevice を実行します。

    spmd
    gpuDevice
end

    parforやparfevalなどの並列言語機能を使用して、並列プールのワーカーに計算を分散します。計算にgpuArray対応関数を使用すると、これらの関数はワーカーの GPU で実行されます。詳細については、GPU での MATLAB 関数の実行を参照してください。例は、複数の GPU での MATLAB 関数の実行を参照してください。

    計算が完了したら、並列プールをシャットダウンします。関数gcpを使用して、現在の並列プールを取得できます。

    delete(gcp("nocreate"));

    他の GPU を選択して使用する場合は、gpuDevice で GPU デバイス インデックスを使用して、各ワーカーで特定の GPU を選択できます。関数 gpuDeviceCount を使用すると、システム内の各 GPU デバイスのインデックスを取得できます。

    システム内に使用可能な GPU が 3 つあり、その中の 2 つだけを計算に使用するとします。デバイスのインデックスを取得します。

    [availableGPUs,gpuIndx] = gpuDeviceCount("available")
    availableGPUs = 
3

    gpuIndx = 1×3

    1    2    3

    使用するデバイスのインデックスを定義します。

    useGPUs = [1 3];

    並列プールを起動します。spmd ブロックと gpuDevice を使用して、各ワーカーと使用する GPU のいずれかをデバイス インデックスにより関連付けます。関数 spmdIndex は各ワーカーのインデックスを識別します。 

    parpool("Processes",numel(useGPUs));
    Starting parallel pool (parpool) using the 'Processes' profile ...
Connected to the parallel pool (number of workers: 2).

    spmd
    gpuDevice(useGPUs(spmdIndex));
end

    ベスト プラクティスとして、また最良のパフォーマンスを得るには、各ワーカーに異なる GPU を割り当てます。

    計算が完了したら、並列プールをシャットダウンします。

    delete(gcp("nocreate"));

    入力引数

    すべて折りたたむ

    デバイスのカウント モード。次のいずれかとして指定します。

    • "all" — GPU デバイス ドライバーによって報告された、すべての GPU デバイスをカウントします。このカウントには、MATLAB でサポートされていないデバイスや、現在の MATLAB セッションで使用できないデバイスも含まれます。

    • "supported" — 現在の MATLAB のバージョンでサポートされている GPU デバイスのみをカウントします。

    • "available" — 現在の MATLAB セッションで使用可能な GPU デバイスのみをカウントします。

    例: "available"

    データ型: char | string

    出力引数

    すべて折りたたむ

    GPU デバイスの数。正のスカラーとして返されます。

    GPU デバイスのインデックス。数値ベクトルとして返されます。indx の各要素は、カウントされた GPU デバイスのデバイス インデックスです。関数 gpuDevice を使用して GPU デバイスを選択またはクエリするには、デバイス インデックスを使用します。

    ヒント

    • 個々のクラスター ワーカーに存在する GPU デバイスの数を返すには、クラスターを使用して並列プールを起動し、spmd ブロックの中で gpuDeviceCount を呼び出します。

      cluster = parcluster("myRemoteGPUCluster");
pool = parpool(cluster);

spmd
  n = gpuDeviceCount
end

      リモート GPU の使用の詳細については、Work with Remote GPUsを参照してください。

    拡張機能

    すべて展開する

    バージョン履歴

    R2010b で導入

    参考

    | | | | |

    トピック