torch.cuda.get_device_name

torch.cuda.get_device_name(device=None)

獲取裝置的名稱。

引數

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

返回

裝置名稱

返回型別

str