torch.cuda.get_device_name

torch.cuda.get_device_name(device=None)

獲取裝置的名稱。

引數

device (torch.device 或 int 或 str, 可選) – 指定返回名稱的裝置。如果此引數為負整數，此函式將不起作用。如果 device 為 None（預設值），則使用由 current_device() 指定的當前裝置。

返回值

裝置的名稱

返回型別

str