目錄

快捷方式

torch.cuda.get_device_capability¶

torch.cuda.get_device_capability(device=None)[source][source]¶

獲取裝置的 cuda 能力。

引數: device (torch.device 或 int 或 str, 可選) – 指定返回裝置能力的裝置。如果此引數為負整數，此函式不執行任何操作 (no-op)。如果 device 為 None（預設值），則使用由 current_device() 給定的當前裝置。
返回: 裝置的 cuda 主要和次要能力
返回型別: tuple(int, int)

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲取問題答案

檢視資源