torch.cuda.get_device_capability

torch.cuda.get_device_capability(device=None)

取得裝置的 CUDA 計算能力。

參數

device (torch.device 或 int 或 str, 選用) – 要回傳計算能力的裝置。 如果此參數為負整數，則此函數不執行任何操作。 如果 device 為 None (預設)，則它會使用由 current_device() 給定的目前裝置。

回傳

裝置的主要和次要 CUDA 計算能力

回傳類型

tuple(int, int)