for obj in list(torch.cuda.memory_allocated()): 提示错误TypeError: 'int' object is not iterable

这个错误提示表明你试图迭代一个 int 对象，而 int 对象不支持迭代操作。推测你的代码应该是这样的：

import torch

memory_usage = torch.cuda.memory_allocated()
for obj in list(memory_usage):
    # do something

`torch.cuda.memory_allocated()` 返回的是一个 int 类型的值，它表示当前 PyTorch 程序在 GPU 上已经分配的内存大小，因此不能直接迭代。如果你想要迭代一个列表，可以将它封装到一个列表中，例如：

import torch

memory_usage = [torch.cuda.memory_allocated()]
for obj in memory_usage:
    # do something

或者直接使用列表：

import torch

memory_usage = [1, 2, 3, 4]  # 这里假设你有一个列表需要迭代
for obj in memory_usage:
    # do something

希望这能帮助你解决问题。