import torch# 检查 CUDA 是否可用print(torch.cuda.is_available())if torch.cuda.is_available():# 获取可用的 CUDA 设备数量print("可用的 CUDA 设备数量:", torch.cuda.device_count())# 获取当前使用的 CUDA 设备索引print("当前 CUDA 设备索引:", torch.cuda.current_device())# 获取指定索引 CUDA 设备的名称print("当前 CUDA 设备名称:", torch.cuda.get_device_name(0))# 创建一个张量并将其移动到 CUDA 设备上x = torch.tensor([1.0,2.0]).cuda()y = torch.tensor([3.0,4.0]).cuda()z = x + yprint("在 CUDA 上计算的结果:", z)else:print("CUDA 不可用")