检测torch-gpu是否安装成功

要测是否成功安装了 torch-gpu，可以通过以下步骤进行验证：

1. 首先，确保已经安装了 CUDA 和对应的 GPU 驱动程序。torch-gpu 需要依赖 CUDA 来加速计算。

2. 在 Python 环境中导入 torch 模块，并查看是否能够成功导入。可以使用以下代码进行导入：

import torch

如果成功导入了 torch 模块，说明 torch-gpu 安装成功。

3. 检查是否能够使用 GPU 运行。通过以下代码可以查看当前系统中是否有可用的 GPU 设备：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)

如果输出结果显示为 "cuda"，则说明成功检测到可用的 GPU 设备，并且 torch-gpu 安装成功