使用pytorch时，如何查看GPU使用情况？

在PyTorch中，你可以使用`torch.cuda.memory_summary()`函数或者直接通过命令行工具来查看GPU的使用情况。以下是两种常用的方法：

1. **代码查看**:
- 首先，你需要导入`torch.cuda`模块，并确保你的设备已经设置好CUDA。

   import torch
   if torch.cuda.is_available():
       print("Current device:", torch.cuda.current_device())
       summary = torch.cuda.memory_summary(device=torch.device('cuda'), abbreviated=True)
       print(summary)

这会打印出当前GPU的内存占用概览。

2. **命令行查看**:
如果你正在使用Jupyter Notebook，可以在终端中运行以下命令：

   nvidia-smi

这将显示所有连接到系统的GPU的信息，包括每个GPU的利用率、内存使用量等。