システム／GPU関連(System/GPU)

その他システム制御関連のAPIです

binarybrain.system.get_cuda_driver_version()

CUDAドライババージョンの取得

戻り値

CUDAドライババージョン

戻り値の型

driver_version (int)

binarybrain.system.get_cuda_driver_version_string()

CUDAドライババージョン文字列の取得

戻り値

CUDAドライババージョン文字列

戻り値の型

driver_version (str)

binarybrain.system.get_device_count()

利用可能なデバイス(GPU)の個数を確認

戻り値

利用可能なデバイス(GPU)の個数を返す

戻り値の型

device_count (int)

binarybrain.system.get_device_properties_string(device_id)

現在のデバイス(GPU)の情報を入れた文字列を取得

パラメータ

device_id (int) -- 情報を取得するデバイス番号を指定

戻り値

現在のデバイス(GPU)の情報を入れた文字列を返す

戻り値の型

device_properties_string (str)

binarybrain.system.get_version_string()

バージョン文字列取得

戻り値

バージョン文字列

戻り値の型

version (str)

binarybrain.system.is_device_available()

デバイス(GPU)が有効かの確認

戻り値

デバイス(GPU)が利用可能なら True を返す

戻り値の型

device_available (bool)

binarybrain.system.omp_set_num_threads(threads: int)

omp_set_num_threadsを呼び出す バックグランドで学習する場合など、Host側のCPUをすべて使うと 逆に性能が落ちる場合や、運用上不便なケースなどで個数制限できる

パラメータ

threads (int) -- OpenMPでのスレッド数

binarybrain.system.set_device(device_id)

利用するデバイス(GPU)を切り替え

パラメータ

device_id (int) -- 利用するデバイス番号を指定

binarybrain.system.set_host_only(host_only: bool)

ホスト(CPU)のみの指定

True を設定するとデバイス(GPU)を未使用としてホスト(CPU)のみを利用

パラメータ

host_only (bool) -- ホストのみの場合 True を指定