验证pytorch的gpu版本安装

在PyTorch中验证GPU版本安装通常涉及到以下几个步骤：

1. **导入**：首先，你需要在Python环境中导入`torch`库，这会自动检测可用的硬件资源。

import torch

2. **查看设备**：然后你可以检查`cuda.is_available()`函数，它会返回一个布尔值，如果GPU可用则为True，否则为False。

if torch.cuda.is_available():
    device = torch.device("cuda")  # 如果有GPU，设置默认设备为GPU
else:
    device = torch.device("cpu")   # 否则设为CPU
print(f"是否有可用的GPU: {torch.cuda.is_available()}")

3. **查看GPU信息**：如果你想获取更详细的GPU型号、显存等信息，可以使用`torch.cuda.get_device_name(0)`。

device_name = torch.cuda.get_device_name(0) if torch.cuda.is_available() else "CPU"
print(f"使用的GPU名称: {device_name}")

4. **测试计算**：可以在GPU上运行一个简单的矩阵运算来确认其是否正常工作。

x_gpu = torch.randn(5, 5).to(device)
y_gpu = x_gpu * 2
print("GPU运算示例:", y_gpu)