PyTorch CUDA get_device_name 関数でGPUデバイスの名前を取得する方法

PyTorchのCUDAにおけるtorch.cuda.get_device_name関数について

概要

• 目的: 利用可能なGPUデバイスの名前を取得する
• 引数:
• 戻り値:
  • 取得したデバイスの名前。文字列型で返却

使用例

import torch

# 利用可能なGPUデバイスの数を取得
num_devices = torch.cuda.device_count()

# 各デバイスの名前を取得
for i in range(num_devices):
    device_name = torch.cuda.get_device_name(i)
    print(f"デバイス{i}の名前: {device_name}")

実行結果

デバイス0の名前: GeForce RTX 3090
デバイス1の名前: GeForce RTX 3080 Ti

補足

• 複数のGPUデバイスを搭載している場合は、device_idを指定することで個別に名前を取得できます。
• デバイス名の取得には、CUDAドライバとcudnnライブラリのインストールが必要です。
• デバイス名が取得できない場合は、エラーが発生します。

torch.cuda.get_device_name関数は、PyTorchのCUDA機能で利用可能なGPUデバイスの名前を取得するための関数です。複数のGPUデバイスを搭載している環境で、個別にデバイスの名前を取得する際に役立ちます。

PyTorch CUDA get_device_name サンプルコード

利用可能なGPUデバイスの名前を取得

import torch

# 利用可能なGPUデバイスの数を取得
num_devices = torch.cuda.device_count()

# 各デバイスの名前を取得
for i in range(num_devices):
    device_name = torch.cuda.get_device_name(i)
    print(f"デバイス{i}の名前: {device_name}")

特定のGPUデバイスの名前を取得

import torch

device_id = 1 # 取得したいデバイスのID

device_name = torch.cuda.get_device_name(device_id)

print(f"デバイス{device_id}の名前: {device_name}")

デバイス名と属性を取得

import torch

# 利用可能なGPUデバイスの数を取得
num_devices = torch.cuda.device_count()

# 各デバイスの名前と属性を取得
for i in range(num_devices):
    device_name = torch.cuda.get_device_name(i)
    device_properties = torch.cuda.get_device_properties(i)
    print(f"デバイス{i}の名前: {device_name}")
    print(f"デバイス{i}の属性: {device_properties}")

デフォルトのデバイスの名前を取得

import torch

device_name = torch.cuda.get_device_name()

print(f"デフォルトデバイスの名前: {device_name}")

CUDAが無効な場合

import torch

try:
    device_name = torch.cuda.get_device_name()
except RuntimeError:
    print("CUDAが無効です")

補足

• 上記のサンプルコードは、Python 3.7以降とPyTorch 1.8以降で動作確認しています。
• CUDAドライバとcudnnライブラリのインストールが必要です。

PyTorch CUDA get_device_name 以外の方法

nvidia-smi コマンド

nvidia-smi --query-gpu=name

このコマンドは、システム上のすべてのGPUデバイスの名前を出力します。

os.uname() 関数

import os

device_name = os.uname().nodename

この方法は、Linux環境でのみ動作します。

platform モジュール

import platform

device_name = platform.node()

この方法は、すべてのプラットフォームで動作しますが、必ずしもGPUデバイスの名前を取得できるとは限りません。

psutil モジュール

import psutil

for gpu in psutil.gpus():
    device_name = gpu.name
    print(f"デバイスの名前: {device_name}")

この方法は、すべてのプラットフォームで動作しますが、psutilモジュールのインストールが必要です。

torch.cuda.get_device_name 以外にも、いくつかの方法でPyTorchで利用可能なGPUデバイスの名前を取得できます。

どの方法を選択するかは、環境や目的に応じて決定してください。

補足

• 上記の方法で取得できるデバイス名は、環境によって異なる場合があります。
• デバイス名が取得できない場合は、エラーが発生する可能性があります。