怎么看pytorch是gpu版本的

PyTorch是一个深度学习框架，它支持在GPU上运行计算，以加速处理大规模数据。查看Python环境中PyTorch是否为GPU版本，你可以通过以下步骤：

1. 打开命令行终端或者Jupyter Notebook。
2. 导入`torch`库（PyTorch的核心模块），然后检查是否有`cuda`或`torch.cuda`模块存在：

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

如果`is_available()`返回`True`，则表示你当前安装的是GPU版本。

3. 你还可以查看设备信息：

   devices = torch.cuda.device_count()
   print(f"可用的CUDA设备数: {devices}")
   device = torch.cuda.get_device_name(0)  # 获取默认设备名称
   print(f"当前使用的设备: {device}")

如果`torch.cuda.is_available()`返回`False`，那就说明你正在使用的不是GPU版本的PyTorch，而是CPU版本。