如何列出所有PyTorch能用的GPU

要查看PyTorch支持的GPU列表，您可以使用以下代码：

首先，确保您已经安装了PyTorch。您可以通过在命令行中输入以下命令来检查PyTorch是否已安装：

import torch
print(torch.__version__)

如果PyTorch已安装，将显示PyTorch的版本号。

接下来，要查看PyTorch支持的GPU列表，您可以使用以下代码：

import torch
# 检查CUDA是否可用
if torch.cuda.is_available():
    print('CUDA is available!')
    # 获取GPU数量
    num_gpus = torch.cuda.device_count()
    print(f'Number of GPUs available: {num_gpus}')
    # 获取每个GPU的型号和索引
    for i in range(num_gpus):
        gpu_name = torch.cuda.get_device_name(i)
        gpu_index = i + 1
        print(f'GPU {gpu_index}: {gpu_name}')
else:
    print('CUDA is not available. Please make sure you have a CUDA-capable GPU and the correct driver and CUDA toolkit installed.')

这段代码将检查CUDA是否可用，并输出可用的GPU数量和型号。如果CUDA不可用，将显示相应的错误消息。

请注意，要使用GPU进行训练，您需要确保安装了正确版本的CUDA工具包，并且您的系统与GPU兼容。在某些情况下，您可能还需要更新您的GPU驱动程序。