torch.cuda.get_device_name

torch.cuda.get_device_name(device=None)

取得裝置的名稱。

參數

device (torch.device 或 int 或 str, optional) – 要返回名稱的裝置。如果此引數為負整數，則此函式不執行任何操作。如果 device 為 None (預設)，則使用 current_device() 提供的目前裝置。

回傳

裝置的名稱

回傳類型

str