如何查看显存

在大多数的操作系统和编程语言中，可以使用相应的函数或命令来查看显存的信息。

在 Python 中，可以使用 PyTorch 库来查看显存信息。具体的代码如下：

import torch
print(torch.cuda.memory_allocated())
print(torch.cuda.memory_cached())

第一行代码可以查看当前已经分配的显存大小，第二行代码可以查看当前已经缓存的显存大小。这些信息可以帮助我们更好地管理显存，防止显存溢出的问题。