根据这个pytorch文档,命令torch.cuda.memory_stats(device="cuda")和torch.cuda.memory_snapshot()应该返回一个dict和一个包含内容的列表。然而,第一个给出了OrderedDict(),第二个给出了[]为什么会这样呢?torch.cuda.is_available()返回True
torch.cuda.memory_stats(device="cuda")
torch.cuda.memory_snapshot()
OrderedDict()
[]
torch.cuda.is_available()
True
qncylg1j1#
因为你没有使用gpu内存,例如,尝试启动一个内核并实现以下代码:
import torch print(torch.cuda.memory_stats(device="cuda")) print(torch.cuda.memory_snapshot()) x = torch.randn(10, 10).to('cuda') print(torch.cuda.memory_stats(device="cuda")) print(torch.cuda.memory_snapshot())
第一个会给予你空的字典和列表,而最后一个会给你想要的静态
1条答案
按热度按时间qncylg1j1#
因为你没有使用gpu内存,例如,尝试启动一个内核并实现以下代码:
第一个会给予你空的字典和列表,而最后一个会给你想要的静态