Python get_device_capability Beispiele

Beispiel #1

def half_supported():
    """
    Returns whether FP16 is support on the GPU
    """
    try:
        return get_device_capability()[0] >= 7
    except:
        return False

Beispiel #2

def half_supported():
    """
    Returns whether FP16 is support on the GPU
    """
    return get_device_capability()[0] >= 7

Beispiel #3

Datei: Daily1022_pytorch（二十二）：torch.cuda：CUDA第二部分.py Projekt: rinetd/TorchDaily

#%%
index = cuda.current_device()
index

#%% [markdown]
# - torch.cuda.device_count():返回当前可用的GPU数目

#%%
num = cuda.device_count()
num

#%% [markdown]
# - torch.cuda.get_device_capability(device): 返回设备的cuda能力

#%%
cuda.get_device_capability(0)

#%% [markdown]
# - torch.cuda.get_device_name(device):返回设备名称

#%%
cuda.get_device_name(0)

#%% [markdown]
# - torch.cuda.max_memory_allocated(device):返回指定设备张量的最大GPU内存用量

#%%
cuda.max_memory_allocated(0)

#%%
device = torch.device('cuda') if cuda.is_available() else torch.device('cpu')