MATLAB での GPU の利用

2021/05/21 作成

csgs2 システムでは、GPU (NVIDIA TESLA V100 GPU) を 搭載しており、これを使用した計算が可能です。

ここでは、MATLAB から GPU を使用して計算する方法をご紹介します。

Parallel Computing Toolbox

MATLAB の toolbox のひとつである Parallel Computing Toolbox を使うことで、GPU を利用することができます。

例えば、以下のように gpuArray() 関数を使うことで、 GPU 上にデータを送ることができます。

N = 6;

M = magic(N);

G = gpuArray(M) ;

G に対して関数を実行することで、GPU を使って計算することができます。 利用できる関数一覧は以下のページに掲載されています。

Use gpuArray Data

そのほか、Parallel Computing Toolbox で GPU を利用する方法については以下のページをご覧下さい

GPU Computing (Parallel Computing Toolbox)