检查 CUDA 是否可用：
使用 torch.cuda.is_available() 来检查 CUDA 是否可用。如果返回 True，则表示 CUDA 可用，并且 PyTorch 能够使用 GPU。

import torch
print(torch.cuda.is_available())

查看可用的 GPU 数量：
使用 torch.cuda.device_count() 可以查看系统中有多少个 GPU 可用。

print(torch.cuda.device_count())

获取 GPU 的详细信息：
如果你想获取每个 GPU 的详细信息，比如名称和 CUDA 版本，你可以使用以下代码：

for i in range(torch.cuda.device_count()):
    print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
    print(f"CUDA Version: {torch.version.cuda}")